Update app.py

app.py

@@ -6,14 +6,15 @@ import gradio as gr
 import os
 
 MODEL_NAME = os.getenv('MODEL_ID')
-
+TOKEN = os.getenv('TOKEN')
 print("Loading model...")
-tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, trust_remote_code=True)
+tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, trust_remote_code=True, token=TOKEN)
 model = AutoModelForCausalLM.from_pretrained(
     MODEL_NAME,
     torch_dtype=torch.bfloat16,
     device_map="auto",
-    trust_remote_code=True
+    trust_remote_code=True,
+    token=TOKEN
 )
 print("Model loaded.")