Update app.py

app.py

@@ -49,21 +49,20 @@ def process_image(img):
     wiki_url = search_terms_wikipedia.get(predicted_class, "No Wikipedia entry found.")
     
     # Generate FLUX image
-    with gr.Status("Generating AI image based on classification..."):
-        result = fal_client.subscribe(
-            "fal-ai/flux/schnell",
-            arguments={
-                "prompt": f"A detailed, artistic interpretation of {predicted_class} flower in natural setting",
-                "image_size": "256x256"  # Low res for testing
-            },
-            with_logs=True,
-            on_queue_update=on_queue_update,
-        )
-        
-        # Convert the image data
-        image_data = base64.b64decode(result['image'])
-        generated_image = Image.open(io.BytesIO(image_data))
-        
+    result = fal_client.subscribe(
+        "fal-ai/flux/schnell",
+        arguments={
+            "prompt": f"A detailed, artistic interpretation of {predicted_class} flower in natural setting",
+            "image_size": "256x256"  # Low res for testing
+        },
+        with_logs=True,
+        on_queue_update=on_queue_update,
+    )
+    
+    # Convert the image data
+    image_data = base64.b64decode(result['image'])
+    generated_image = Image.open(io.BytesIO(image_data))
+    
     return classification_results, generated_image, wiki_url
 
 # Load the learner