camenduru commited on
Commit
984fef6
·
verified ·
1 Parent(s): 1228077

Update worker_runpod.py

Browse files
Files changed (1) hide show
  1. worker_runpod.py +2 -3
worker_runpod.py CHANGED
@@ -176,7 +176,7 @@ def generate(input):
176
 
177
  output_video, output_model_obj = make3d(mv_images, model, device, IS_FLEXICUBES, infer_config, export_video, export_texmap)
178
  mesh_basename = os.path.splitext(output_model_obj)[0]
179
- result = output_video, [output_model_obj, mesh_basename+'.mtl', mesh_basename+'.png']
180
 
181
  response = None
182
  try:
@@ -187,8 +187,7 @@ def generate(input):
187
  job_id = values['job_id']
188
  del values['job_id']
189
 
190
- first_key = next(iter(result[0]))
191
- file_path = result[0][first_key]
192
  file_paths = result[1]
193
  default_filename = os.path.basename(file_path)
194
  files = { default_filename: open(file_path, "rb").read() }
 
176
 
177
  output_video, output_model_obj = make3d(mv_images, model, device, IS_FLEXICUBES, infer_config, export_video, export_texmap)
178
  mesh_basename = os.path.splitext(output_model_obj)[0]
179
+ result = [output_video, [output_model_obj, mesh_basename+'.mtl', mesh_basename+'.png']]
180
 
181
  response = None
182
  try:
 
187
  job_id = values['job_id']
188
  del values['job_id']
189
 
190
+ file_path = result[0]
 
191
  file_paths = result[1]
192
  default_filename = os.path.basename(file_path)
193
  files = { default_filename: open(file_path, "rb").read() }