Update app.py

app.py

@@ -132,7 +132,7 @@ def image_to_3d(trial_id: str, seed: int, randomize_seed: bool, ss_guidance_stre
             "cfg_strength": slat_guidance_strength,
         },
     )
-    video = render_utils.render_video(outputs['gaussian'][0], num_frames=120)['color']
+    video = render_utils.render_video(outputs['mesh'][0], num_frames=120)['color']
     video_geo = render_utils.render_video(outputs['mesh'][0], num_frames=120)['normal']
     video = [np.concatenate([video[i], video_geo[i]], axis=1) for i in range(len(video))]
     trial_id = uuid.uuid4()
@@ -214,8 +214,8 @@ with gr.Blocks() as demo:
     with gr.Row():
         examples = gr.Examples(
             examples=[
-                f'assets/example_image/{image}'
-                for image in os.listdir("assets/example_image")
+                f'assets/example_imgs_trellis_oa/{image}'
+                for image in os.listdir("assets/example_imgs_trellis_oa")
             ],
             inputs=[image_prompt],
             fn=preprocess_image,
@@ -266,7 +266,7 @@ with gr.Blocks() as demo:
 
 # Launch the Gradio app
 if __name__ == "__main__":
-    pipeline = TrellisImageTo3DPipeline.from_pretrained("JeffreyXiang/TRELLIS-image-large")
+    pipeline = TrellisImageTo3DPipeline.from_pretrained("Louischong/Trellis-OA")
     if torch.cuda.is_available():
         pipeline.cuda()
         print("CUDA is available. Using GPU.")