Update app.py

app.py

@@ -5,7 +5,7 @@ from llama_mesh import LLaMAMesh
 from mast3r import MASt3R
 from utils import apply_gradient_color
 from utils import create_image_grid
-
+import os
 import torch
 
 DESCRIPTION = '''
@@ -16,10 +16,11 @@ DESCRIPTION = '''
 '''
 
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
+HF_TOKEN = os.environ.get("HF_TOKEN", None)
 
 # Initialize models
 mv_diff = MultiViewDiffusion(device=DEVICE)
-vllm = VisionLLM(device=DEVICE)
+vllm = VisionLLM(device=DEVICE, use_auth_token=HF_TOKEN)
 llama_mesh = LLaMAMesh(device=DEVICE)
 mast3r = MASt3R(device=DEVICE)
 
@@ -95,6 +96,4 @@ with gr.Blocks() as demo:
           fn=apply_gradient_color,
           inputs=[future_mesh_input],
            outputs=[future_mesh_output_2]
-        )
-
-demo.launch()
+        )