Update app.py

app.py

@@ -1189,7 +1189,16 @@ def generate_image_flux(prompt):
         num_inference_steps=4,
         api_name="/infer"
     )
-    return result
+    
+    # Assuming that the API response contains an image file or URL, extract the image part
+    if isinstance(result, tuple):
+        # Extract the image URL or path if it is a tuple
+        image_path_or_url = result[0]  # Adjust this index based on the actual structure of the response
+    else:
+        image_path_or_url = result
+    
+    return image_path_or_url  # Return the image path or URL directly
+
 
 # Function to update images with the three prompts
 def update_images():