Update app.py

app.py

@@ -1,10 +1,12 @@
 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
+import os
+
 
 # Cargar el modelo y el tokenizer
 model_id = "meta-llama/Llama-3.2-1B"
-tokenizer = AutoTokenizer.from_pretrained(model_id)
-model = AutoModelForCausalLM.from_pretrained(model_id)
+tokenizer = AutoTokenizer.from_pretrained(model_id, use_auth_token=os.getenv("TOKEN"))
+model = AutoModelForCausalLM.from_pretrained(model_id, use_auth_token=os.getenv("TOKEN"))
 
 def generate_text(prompt):
     inputs = tokenizer(prompt, return_tensors="pt")