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
-    result = client.predict(image_path, prompt, api_name="/predict")
-    return result
+    (padded_image, images.images[0]), padded_image, anyline_image = client.predict(image_path, prompt, api_name="/predict")
+    return anyline_image
 
 # Create the Gradio interface
 iface = gr.Interface(