Update app.py

app.py

@@ -1,22 +1,21 @@
 import gradio as gr
 from PIL import Image
 
-# Directory containing local images
-IMAGE_DIR = "images/"
+# Dictionary mapping animals to image file names in the root directory
+animal_images = {
+    "cat": "cat.jpg",
+    "dog": "dog.jpg",
+    "bird": "bird.jpg",
+    "horse": "horse.jpg",
+    # Add more animals and image filenames as needed
+}
 
 def fetch_animal_image(animal):
-    animal_images = {
-        "cat": "cat.jpg",
-        "dog": "dog.jpg",
-        "bird": "bird.jpg",
-        "horse": "horse.jpg",
-    }
-    
     # Fetch image file name based on user input
     image_file = animal_images.get(animal.lower(), "default.jpg")  # Fallback image
     
-    # Open and return the image
-    img_path = IMAGE_DIR + image_file
+    # Open and return the image from the root directory
+    img_path = image_file
     img = Image.open(img_path)
     
     return img
@@ -32,3 +31,4 @@ demo = gr.Interface(
 
 # Launch the interface
 demo.launch()
+