Update app.py

app.py

@@ -59,12 +59,12 @@ def get_results(image, prompt):
 
     
 # Define the input components for Gradio (adding a new input for the prompt)
-image_input = gr.inputs.Image()
-text_input = gr.inputs.Textbox(label="Prompt")  # New input for the text prompt
+# image_input = gr.inputs.Image()
+# text_input = gr.inputs.Textbox(label="Prompt")  # New input for the text prompt
 
 
-# Define the output components for Gradio (including both image and text)
-outputs = gr.Image(type="numpy", label="Output Image")
+# # Define the output components for Gradio (including both image and text)
+# outputs = gr.Image(type="numpy", label="Output Image")
 
 # Define the text description within an HTML <div> element
 description_html = """
@@ -105,16 +105,29 @@ title = "autoannotation"
 
 description = "This is a project description. It demonstrates how to use Gradio with an image and text input to interact with an API."
 
+import os
+examples = [
+    ["3000.jpeg",'person,car,traffic sign,traffic light'],
+    ["original (8).jpg", 'person,car,traffic sign,traffic light'],
+    ["traffic.jpg", 'person,car,traffic sign,traffic light'],
+]
+
+
+
+
 # Create a Blocks object and use it as a context manager
 with gr.Blocks() as demo:
     gr.Markdown(
             """
             # Tuba Autoannotation Demo
-            This is your private demo for [Tuba Autoannotation](https://Tuba.ai),
-            a simple and controllable model for music generation
+
+            This is your private demo for [Tuba Autoannotation](https://Tuba.ai)
+
+            A prompt based controllable model for auto annotation
             """
         )
     # Define the input components and add them to the layout
+    
     with gr.Row():
         image_input = gr.inputs.Image()
         output = gr.Image(type="numpy", label="Output Image")
@@ -128,6 +141,12 @@ with gr.Blocks() as demo:
     # Define the event listener that connects the input and output components and triggers the function
     button.click(fn=get_results, inputs=[image_input, text_input], outputs=output, api_name="get_results")
     # Add the description below the layout
+    gr.Examples(
+            fn=get_results,
+            examples=examples,
+            inputs=[image_input, text_input],
+            outputs=[outputs]
+        )
     gr.Markdown(description_html)
 # Launch the app
 demo.launch(share=False)