Update app.py

app.py

@@ -7,12 +7,9 @@ import textwrap
 #model = "mistralai/Mixtral-8x7B-Instruct-v0.1"
 
 
-from peft import PeftModel, PeftConfig
-from transformers import AutoModelForCausalLM
-
-config = PeftConfig.from_pretrained("TachyHealthResearch/Mistral-7B-Medical-Finetune_V2")
-base_model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-Instruct-v0.1")
-model = PeftModel.from_pretrained(base_model, "TachyHealthResearch/Mistral-7B-Medical-Finetune_V2")
+# Load model directly
+from transformers import AutoModel
+model = AutoModel.from_pretrained("GRMenon/mental-health-mistral-7b-instructv0.2-finetuned-V2")
 client = InferenceClient(model)
 
 # Embedded system prompt