sds

app.py

@@ -30,7 +30,7 @@ def create_from_text(prompt):
     subprocess.run(cmd1)
     subprocess.run(cmd2)
     glb_path = convert_obj_to_glb(f"{temp_dir}/{sanitized_prompt}.obj", temp_dir)
-    return get_html_model(glb_path)
+    return get_html_model(glb_path), glb_path
     
 def create_from_image(image):
     temp_dir = tempfile.mkdtemp()
@@ -65,7 +65,7 @@ def create_from_image(image):
     subprocess.run(cmd2)
     subprocess.run(cmd3)
     glb_path = convert_obj_to_glb(f"{temp_dir}/{sanitized_prompt}.obj", temp_dir)
-    return get_html_model(glb_path)
+    return get_html_model(glb_path), glb_path
 
 
 def convert_obj_to_glb(obj_path, out_path):
@@ -105,7 +105,8 @@ source: https://github.com/dreamgaussian/dreamgaussian
                 btn = gr.Button("Generate")
         with gr.Column():
             model_3d = gr.HTML(label="Model 3D", show_label=True)
-    btn.click(generate, inputs=[prompt, image], outputs=[model_3d])
+            file = gr.File()
+    btn.click(generate, inputs=[prompt, image], outputs=[model_3d, file])
     clear.click(lambda x: (gr.update(value=None),gr.update(value=None)), None, [prompt, image], queue=False)
 
 demo.queue(api_open=False, concurrency_count=1)