Update app.py

app.py

@@ -1,12 +1,13 @@
+import os
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 
 # Use the base (untrained) model from Hugging Face Hub
 model_id = "mistralai/Mistral-7B-Instruct-v0.3"
-  # Your Hugging Face token
+api_key = os.environ.get("API_TOKEN")  # Your Hugging Face token
 
-tokenizer = AutoTokenizer.from_pretrained(model_id)
-model = AutoModelForCausalLM.from_pretrained(model_id)
+tokenizer = AutoTokenizer.from_pretrained(model_id, token = api_key)
+model = AutoModelForCausalLM.from_pretrained(model_id, token = api_key)
 
 pipe = pipeline(
     "text-generation",