Update app.py

app.py

@@ -1,10 +1,14 @@
 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
+from huggingface_hub import login
+
+# Authenticate using token
+login(token="your_token_here")
 
 # Load the model and tokenizer
 model_name = "mistralai/Mistral-Nemo-Instruct-2407"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForCausalLM.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=True)
+model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=True)
 
 def generate_response(prompt):
     inputs = tokenizer(prompt, return_tensors="pt")