Update app.py

app.py

@@ -47,7 +47,9 @@ def extract_medicines(image_files):
     if not image_files:
         return "Please upload a prescription image."
     
-    images = [load_image(image) for image in image_files]
+    # Handle file inputs correctly
+    image_paths = [file.name for file in image_files] if isinstance(image_files, list) else [image_files.name]
+    images = [load_image(path) for path in image_paths]
     
     # Specific prompt to extract only medicine names
     text = "Extract ONLY the names of medications/medicines from this prescription image. Format the output as a numbered list of medicine names only, without dosages or instructions."
@@ -106,22 +108,15 @@ with gr.Blocks() as demo:
         outputs=output
     )
     
-    gr.Examples(
-        examples=[
-            ["examples/prescription1.jpg"],
-            ["examples/prescription2.jpg"],
-        ],
-        inputs=image_input,
-        outputs=output,
-        fn=extract_medicines,
-        cache_examples=True,
-    )
+    # Note: For examples to work with current Gradio versions, you need a different approach
+    # than what I previously provided. Remove examples for now to fix the immediate error.
     
     gr.Markdown("""
     ### Notes:
     - This app is optimized to run on CPU
     - Upload clear images of prescriptions for best results
     - Only medicine names will be extracted
+    - Processing might take a minute or two on CPU
     """)
 
 demo.queue()