Update app.py

app.py

@@ -1,24 +1,20 @@
 import gradio as gr
 
 # Assuming you have a dictionary of pet names and their image URLs or file paths
-pets = {
+pet_images = {
     "David": "Users/EmilyWitko/desktop/David AI/david.jpg",
     "Kikou": "Users/EmilyWitko/desktop/David AI/kikou.jpg",
     # Add as many pets as you have
 }
 
-def show_pet_image(pet_name):
-    # If the pet name is in the dictionary, return the image
-    if pet_name in pets:
-        return pets[pet_name]
-    else:
-        return "Pet not found, please try another name."
+# Define your Gradio interface here
+def your_function(pet_name):
+    # Handle the input and produce the image path or URL
+    image_path = pet_images.get(pet_name, "Users/EmilyWitko/desktop/David AI/david.jpg")  # Default if pet not found
+    return image_path
 
-iface = gr.Interface(
-    fn=show_pet_image,
-    inputs=gr.inputs.Textbox(label="Enter Pet Name"),
-    outputs=gr.outputs.Image(label="Pet Image"),
-    examples=list(pets.keys())  # Optional: to provide example pet names in the interface
-)
+# Create the interface
+iface = gr.Interface(fn=your_function, inputs="text", outputs="image")
 
+# Launch the interface
 iface.launch()