Access token implemented

app.py

@@ -1,15 +1,21 @@
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
+import os
 
 MODEL_NAME = "tiiuae/falcon-180B"
 
-tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, trust_remote_code=True)
+access_token = os.getenv("HF_ACCESS_TOKEN")
+tokenizer = AutoTokenizer.from_pretrained(
+    MODEL_NAME, 
+    trust_remote_code=True,
+    use_auth_token=access_token)
 model = AutoModelForCausalLM.from_pretrained(
     MODEL_NAME,
     torch_dtype=torch.bfloat16,
     device_map="auto",
-    trust_remote_code=True
+    trust_remote_code=True,
+    use_auth_token=access_token
 )
 
 def improve_code(code: str) -> str: