Update app.py

app.py

@@ -1,10 +1,13 @@
+import os
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
+access_token = os.environ["GATED_ACCESS_TOKEN"]
+
 # Load the tokenizer and model
 model_id = "mistralai/Mixtral-8x22B-v0.1"
-tokenizer = AutoTokenizer.from_pretrained(model_id)
-model = AutoModelForCausalLM.from_pretrained(model_id)
+tokenizer = AutoTokenizer.from_pretrained(model_id, token=access_token)
+model = AutoModelForCausalLM.from_pretrained(model_id, token=access_token)
 
 # Function to generate text using the model
 def generate_text(prompt, max_length=500, temperature=0.7, top_k=50, top_p=0.95, num_return_sequences=1):