Update app.py

app.py

@@ -60,7 +60,15 @@ def generate_image(prompt, structure_image, style_image, depth_strength=15, styl
         style_strength=style_strength,
         api_name="/generate_image"
     )
-    return result['url']
+
+    if isinstance(result, str) and os.path.exists(result):
+        output_image = Image.open(result)
+    elif isinstance(result, bytes):
+        output_image = Image.open(BytesIO(result))
+    else:
+        raise ValueError(f"Неожиданный тип результата API: {type(result)}")
+
+    return output_image
 
 # Примеры для Gradio
 examples = [