Update worker_runpod.py
Browse files- worker_runpod.py +1 -1
worker_runpod.py
CHANGED
|
@@ -84,7 +84,7 @@ def generate(input):
|
|
| 84 |
supported_tasks[args.task](args).run()
|
| 85 |
|
| 86 |
if args.task == "fr_bg":
|
| 87 |
-
result = "/content/result/
|
| 88 |
else:
|
| 89 |
result = "/content/result/diffbir_tost.png"
|
| 90 |
try:
|
|
|
|
| 84 |
supported_tasks[args.task](args).run()
|
| 85 |
|
| 86 |
if args.task == "fr_bg":
|
| 87 |
+
result = "/content/result/diffbir_tost_0.png"
|
| 88 |
else:
|
| 89 |
result = "/content/result/diffbir_tost.png"
|
| 90 |
try:
|