Update app.py

app.py

@@ -103,7 +103,54 @@ def predict_answer(image, video, question):
     else:
         return "Unsupported file type. Please upload an image or video."
         
-
+promt_cat_dog = """
+        Annotate this image with this schema:
+        {
+        “description”: “Is there a cat in the image?”,
+        “value”: “Cat”
+        },
+        {
+        “description”: “Is there a dog in the image?”,
+        “value”: “Dog”
+        },
+        {
+        “description”: “Is there a horse in the image?”,
+        “value”: “Horse”
+        },
+        
+        provide me the answers as a dictionary with key as the string value of the variable value on top and its value should be 0 if it is Flase, else 1 if its is True
+        """
+promt_bus_people = """
+        Annotate this image with this schema:
+        {
+        “description”: “Is there a bus in the image?”,
+        “value”: “Bus”
+        },
+        {
+        “description”: “Is there a bike in the image?”,
+        “value”: “Bike”
+        },
+        
+        provide me the answers as a dictionary with key as the string value of the variable value on top and its value should be 0 if it is Flase, else 1 if its is True
+        """
+
+promt_video = """
+        Annotate this image with this schema:
+        {
+        “description”: “Is the person standing?”,
+        “value”: “Standing”
+        },
+        {
+        “description”: “Is the person hands free?”,
+        “value”: “Hands-Free”
+        }
+        provide me the answers as a dictionary with key as the string value of the variable value on top and its value should be 0 if it is Flase, else 1 if its is True
+        """
+test_examples = [[None, "Images/cat_dog.jpeg", promt_cat_dog], 
+            [None,"Images/bus_people.jpeg", promt_bus_people], 
+            ["videos/v1.mp4",None,promt_video],
+            ["videos/v2.mp4",None,promt_video],
+            ["videos/v3.mp4",None,promt_video]]
 
 
 def gradio_predict(image, video, question):
@@ -147,5 +194,15 @@ with gr.Blocks(css = css) as app:
     
     btn.click(gradio_predict, inputs=[image, video, question], outputs=answer)
 
+    # Define examples for images
+        gr.Markdown("## Examples")
+    gr.Examples(
+        examples=test_examples,
+        inputs=[image, video, question],
+        outputs=[answer],
+        fn=gradio_predict,
+        cache_examples=True,
+    )
+
 app.launch(debug=True)