Update app.py

app.py

@@ -1,9 +1,14 @@
-
 import gradio as gr
 from typing import Iterator, List, Tuple
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 from peft import PeftConfig, PeftModel
+from huggingface_hub import login
+import os
+
+# Authenticate with Hugging Face
+HUGGINGFACE_TOKEN = os.getenv('HUGGINGFACE_TOKEN')  # Ensure this environment variable is set
+login(token=HUGGINGFACE_TOKEN)
 
 base_model = "mistralai/Mistral-7B-Instruct-v0.2"
 adapter = "GRMenon/mental-health-mistral-7b-instructv0.2-finetuned-V2"