Update app.py

app.py

@@ -40,9 +40,10 @@ def view_random_image(target_dir, target_class):
   target_folder = target_dir+target_class
 
   # Get a random image path
-  random_image = random.choices(os.listdir(target_folder), k=1)
+  random_image = random.sample(os.listdir(target_folder), 1)
     
 
+
   # Read in the image and plot it using matplotlib
   img = mpimg.imread(target_folder + "/" + random_image[0])
   plt.imshow(img)