Spaces:
Running on Zero
Running on Zero
changes
Browse files
app.py
CHANGED
|
@@ -230,20 +230,23 @@ def run_lora(
|
|
| 230 |
print("fatal error", e)
|
| 231 |
image = None
|
| 232 |
|
| 233 |
-
result = {"status": "failed", "message": error_message} if image is None else {"status": "success", "message": "Image generated but not uploaded"}
|
|
|
|
| 234 |
if image is not None and upload_to_r2:
|
| 235 |
try:
|
| 236 |
url = upload_image_to_r2(image, account_id, access_key, secret_key, bucket)
|
| 237 |
-
result = {"status": "success", "message": "upload image success", "url": url}
|
| 238 |
except Exception as e:
|
| 239 |
err = f"Upload failed: {e}"
|
| 240 |
gr.Warning(err)
|
| 241 |
print(err)
|
| 242 |
-
result = {"status": "success", "message": "generated but upload failed"}
|
| 243 |
|
| 244 |
gr.Info("Completed!")
|
| 245 |
progress(100, "Completed!")
|
| 246 |
-
|
|
|
|
|
|
|
| 247 |
|
| 248 |
# Gradio interface
|
| 249 |
|
|
|
|
| 230 |
print("fatal error", e)
|
| 231 |
image = None
|
| 232 |
|
| 233 |
+
result = {"status": "failed", "message": error_message} if image is None else {"status": "success", "message": "Image generated but not uploaded", "seed": seed}
|
| 234 |
+
|
| 235 |
if image is not None and upload_to_r2:
|
| 236 |
try:
|
| 237 |
url = upload_image_to_r2(image, account_id, access_key, secret_key, bucket)
|
| 238 |
+
result = {"status": "success", "message": "upload image success", "url": url, "seed": seed}
|
| 239 |
except Exception as e:
|
| 240 |
err = f"Upload failed: {e}"
|
| 241 |
gr.Warning(err)
|
| 242 |
print(err)
|
| 243 |
+
result = {"status": "success", "message": "generated but upload failed", "seed": seed}
|
| 244 |
|
| 245 |
gr.Info("Completed!")
|
| 246 |
progress(100, "Completed!")
|
| 247 |
+
|
| 248 |
+
# CHANGED: Return both image AND json
|
| 249 |
+
return image, json.dumps(result) # <--- THIS IS THE KEY CHANGE
|
| 250 |
|
| 251 |
# Gradio interface
|
| 252 |
|