TRELLIS_TextTo3D_Try2

Running on Zero

dkatz2391 commited on May 15

Commit

df497d3

verified ·

1 Parent(s): 55c1cbf

cavaraga

Files changed (1) hide show

app.py CHANGED Viewed

@@ -372,7 +372,7 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
 # Launch the Gradio app
 if __name__ == "__main__":
-    pipeline = TrellisTextTo3DPipeline.from_pretrained("JeffreyXiang/TRELLIS-text-xlarge")
     pipeline.cuda()
     # Explicitly allow serving files from the base temp dir and the Gradio cache dir
     allowed_paths = [TMP_DIR, "/tmp/gradio"]

 # Launch the Gradio app
 if __name__ == "__main__":
+    pipeline = TrellisTextTo3DPipeline.from_pretrained("cavargas10/TRELLIS-text-xlarge")
     pipeline.cuda()
     # Explicitly allow serving files from the base temp dir and the Gradio cache dir
     allowed_paths = [TMP_DIR, "/tmp/gradio"]