Update worker_runpod.py

worker_runpod.py

@@ -132,7 +132,7 @@ def make3d(images, model, device, IS_FLEXICUBES, infer_config, export_video, exp
 
 @torch.inference_mode()
 def generate(input):
-    values = json.loads(command)
+    values = input["input"]
     input_image = values['input_image']
     sample_steps = values['sample_steps']
     seed = values['seed']