Reduce images available to 20. Print images. Select 1 random image.

app.py

@@ -18,8 +18,15 @@ from datasets import load_dataset
 dataset = load_dataset("merve/coco")
 
 df = pd.dataset
-print(df.head(12))
+# Select first 20 rows 
+df.iloc[:20,:]
+print ("Print to show the 20 images available.")
+print ("The app will then select an image for further exploration.")
+print(df.head(20))
 
+# Use the sample command 
+selected_image = df.sample(n=1)
+print (selected_image)
 
 # Get image database
 ##curl -X GET \
@@ -33,27 +40,4 @@ print(df.head(12))
 ##model = AutoModelForVision2Seq.from_pretrained("Salesforce/blip-image-captioning-large")
 
 
-import os
-import random
-
-path = "https://huggingface.co/Salesforce/blip-image-captioning-large"
-
-images = []
-
-# This will get each root, dir and file list in the path specified recursively (like the "find" command in linux, but separating files, from directories, from paths).
-# root is the full path from your specified path to the the directory it is inspecting
-# dirs is a list containing all the directories found in the current inspecting directory
-# files is a list containing all the files found in the current inspecting directory
-for root, dirs, files in os.walk(path):
-    # This will filter all the .png files in case there is something else in the directory
-    # If your directory only has images, you can do this:
-    # images = files
-    # instead of filtering the '.png' images with the for loop // filter for jgp
-    for f in files:
-        if f[-4:] == '.jpg':
-            images.append(f)
-
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
-
-print(random.choice(images))
 demo.launch()