Update app.py

app.py

@@ -1,6 +1,5 @@
 # Not ready to use yet
 import spaces
-import base64
 import argparse
 import numpy as np
 import gradio as gr
@@ -96,7 +95,6 @@ def preprocess_image(image, background_choice, foreground_ratio, backgroud_color
     return image.convert("RGB")
 
 @spaces.GPU
-
 def gen_image(input_image, seed, scale, step):
     global pipeline, model, args
     pipeline.set_seed(seed)
@@ -105,16 +103,9 @@ def gen_image(input_image, seed, scale, step):
     stage2_images = rt_dict["stage2_images"]
     np_imgs = np.concatenate(stage1_images, 1)
     np_xyzs = np.concatenate(stage2_images, 1)
-    
+
     glb_path = generate3d(model, np_imgs, np_xyzs, args.device)
-    
-    # Read the GLB file and encode it in base64
-    with open(glb_path, 'rb') as f:
-        glb_bytes = f.read()
-    encoded_glb = 'data:model/gltf-binary;base64,' + base64.b64encode(glb_bytes).decode('utf-8')
-    
-    # Return images and the encoded GLB data
-    return Image.fromarray(np_imgs), Image.fromarray(np_xyzs), encoded_glb
+    return Image.fromarray(np_imgs), Image.fromarray(np_xyzs), glb_path#, obj_path
 
 
 parser = argparse.ArgumentParser()