Update app.py

app.py

@@ -7,9 +7,10 @@ import torch
 from transformers import MllamaForConditionalGeneration, AutoProcessor
 from OmniGen import OmniGenPipeline
 
+Llama32V_HFtoken = os.getenv("Llama32V")
 pipe = OmniGenPipeline.from_pretrained("Shitao/OmniGen-v1")
 model_id = "meta-llama/Llama-3.2-11B-Vision-Instruct"
-model = MllamaForConditionalGeneration.from_pretrained(model_id, torch_dtype=torch.bfloat16, device_map="auto",)
+model = MllamaForConditionalGeneration.from_pretrained(model_id, torch_dtype=torch.bfloat16, device_map="auto",hf_token = Llama32V_HFtoken)
 processor = AutoProcessor.from_pretrained(model_id)
 
 @spaces.GPU()