Update app.py

app.py

@@ -7,10 +7,13 @@ import torch
 from transformers import MllamaForConditionalGeneration, AutoProcessor
 from OmniGen import OmniGenPipeline
 
+from huggingface_hub import login
 Llama32V_HFtoken = os.getenv("Llama32V")
+login(Llama32V_HFtoken)
+
 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",hf_token = Llama32V_HFtoken)
+model = MllamaForConditionalGeneration.from_pretrained(model_id, torch_dtype=torch.bfloat16, device_map="auto")
 processor = AutoProcessor.from_pretrained(model_id)
 
 @spaces.GPU()