Update worker_runpod.py

worker_runpod.py

@@ -207,8 +207,10 @@ def generate(input):
     except Exception as e:
         print(f"An unexpected error occurred: {e}")
     finally:
-        if os.path.exists(result):
-            os.remove(result)
+        if os.path.exists(file_path):
+            os.remove(file_path)
+            for path in file_paths:
+                os.remove(path)
 
     if response and response.status_code == 200:
         try: