Update app.py

app.py

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