Update app.py

app.py

@@ -2,12 +2,14 @@ import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 from peft import PeftModel
 import torch
+import os 
 
 """
 For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
 """
 
 # Set your model and adapter paths
+API_KEY = os.environ.get("ACESS_TOKEN")
 BASE_MODEL = "mistralai/Mistral-7B-v0.1"
 PEFT_ADAPTER = "asdc/Mistral-7B-multilingual-temporal-expression-normalization"
 
@@ -16,6 +18,7 @@ base_model = AutoModelForCausalLM.from_pretrained(
     BASE_MODEL,
     torch_dtype=torch.float16,
     device_map="auto"
+    token=API_KEY
 )
 
 model = PeftModel.from_pretrained(base_model, PEFT_ADAPTER)