Update app.py

app.py

@@ -227,8 +227,8 @@ async def process_image(
         preprocessed = preprocess(image_pil, do_remove_background, foreground_ratio)
         mesh_name_obj, mesh_name_glb = generate(preprocessed, mc_resolution)
         timestamp = datetime.datetime.now().strftime('%Y%m%d%H%M%S%f')
-        object_name = f'object_{timestamp}_1.obj'
-        object_name_2 = f'object_{timestamp}_2.glb'
+        object_name = f'object_{timestamp}.obj'
+        object_name_2 = f'object_{timestamp}.glb'
         object_name_3 = f"object_{timestamp}.png"
         preprocessed_image_tempfile = tempfile.NamedTemporaryFile(suffix=".png", delete=False)
         preprocessed.save(preprocessed_image_tempfile.name)
@@ -238,6 +238,7 @@ async def process_image(
             # torch.cuda.synchronize()  # Wait for all CUDA operations to complete
             # torch.cuda.empty_cache()
             return {
+                "img_path": f"https://framebucket3d.s3.amazonaws.com/{object_name}"
                 "obj_path": f"https://framebucket3d.s3.amazonaws.com/{object_name}",
                 "glb_path": f"https://framebucket3d.s3.amazonaws.com/{object_name_2}"
             }