init

app.py

@@ -23,7 +23,8 @@ from huggingface_hub import snapshot_download
 import os
 local_path = snapshot_download(
     repo_id="KangLiao/Puffin",
-    repo_type="model",                
+    repo_type="model",       
+    filename="Puffin-Base.pth",         
     local_dir="checkpoints/",  
     local_dir_use_symlinks=False,     
     revision="main",                  
@@ -201,7 +202,7 @@ with gr.Blocks(css=css) as demo:
         understanding_button = gr.Button("Chat")
         understanding_output = gr.Textbox(label="Response")
         
-        image_output = gr.Gallery(label="Camera Maps", columns=1, rows=1)
+        camera_output = gr.Gallery(label="Camera Maps", columns=1, rows=1)
 
         with gr.Accordion("Advanced options", open=False):
             und_seed_input = gr.Number(label="Seed", precision=0, value=42)
@@ -224,7 +225,7 @@ with gr.Blocks(css=css) as demo:
     understanding_button.click(
         camera_understanding,
         inputs=[image_input, und_seed_input],
-        outputs=[understanding_output, image_output]
+        outputs=[understanding_output, camera_output]
     )
 
 demo.launch(share=True)