Update app.py

app.py

@@ -1,6 +1,15 @@
 from transformers import AutoTokenizer, AutoModelForCausalLM
+from peft import PeftModel, PeftConfig
+from transformers import AutoModelForCausalLM
+
 import gradio as gr
 
+
+
+config = PeftConfig.from_pretrained("Tonic/mistralmed")
+model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-v0.1")
+model = PeftModel.from_pretrained(model, "Tonic/mistralmed")
+
 # Use the base model's ID
 base_model_id = "mistralai/Mistral-7B-v0.1"
 model_directory = "Tonic/mistralmed"