vibs08 commited on
Commit
530d878
·
verified ·
1 Parent(s): e77b448

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -229,6 +229,10 @@ async def process_image(
229
  timestamp = datetime.datetime.now().strftime('%Y%m%d%H%M%S%f')
230
  object_name = f'object_{timestamp}_1.obj'
231
  object_name_2 = f'object_{timestamp}_2.glb'
 
 
 
 
232
 
233
  if upload_file_to_s3(mesh_name_obj, 'framebucket3d',object_name) and upload_file_to_s3(mesh_name_glb, 'framebucket3d',object_name_2):
234
  # torch.cuda.synchronize() # Wait for all CUDA operations to complete
 
229
  timestamp = datetime.datetime.now().strftime('%Y%m%d%H%M%S%f')
230
  object_name = f'object_{timestamp}_1.obj'
231
  object_name_2 = f'object_{timestamp}_2.glb'
232
+ object_name_3 = f"object_{timestamp}.png"
233
+ preprocessed_image_tempfile = tempfile.NamedTemporaryFile(suffix=".png", delete=False)
234
+ preprocessed.save(preprocessed_image_tempfile.name)
235
+ upload_file_to_s3(preprocessed_image_tempfile.name, 'framebucket3d', object_name_3)
236
 
237
  if upload_file_to_s3(mesh_name_obj, 'framebucket3d',object_name) and upload_file_to_s3(mesh_name_glb, 'framebucket3d',object_name_2):
238
  # torch.cuda.synchronize() # Wait for all CUDA operations to complete