Update worker_runpod.py

worker_runpod.py

@@ -83,7 +83,7 @@ def generate(input):
         }
         supported_tasks[args.task](args).run()
 
-    if args.version == "fr_bg":
+    if args.task == "fr_bg":
         result = "/content/result/restored_backgrounds/diffbir_tost.png"
     else:
         result = "/content/result/diffbir_tost.png"