Update app.py

app.py

@@ -25,11 +25,14 @@ model = AutoModelForCausalLM.from_pretrained(
     return_dict=True,
     torch_dtype=torch.float16,
     revision="main",  # Or the desired revision
+    token=api_token
 )
 
 tokenizer = AutoTokenizer.from_pretrained(
     model_name,
-    revision="main",  # Or the desired revision
+    revision="main", 
+    token=api_token
+    # Or the desired revision
 )
 
 print("loaded model")