Update app.py

app.py

@@ -19,8 +19,8 @@ def enhance_image(image, prompt):
     # Preprocess the image
     image_path = preprocess_image(image)
     # Use the client to predict the result
-    (padded_image, image_1), padded_image, anyline_image = client.predict(file(image_path), prompt, api_name="/predict")
-    return anyline_image
+    (padded_image, image_out), padded_image, anyline_image = client.predict(file(image_path), prompt, api_name="/predict")
+    return image_out
 
 # Create the Gradio interface
 iface = gr.Interface(