Update app.py

app.py

@@ -1,12 +1,13 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
 from transformers import AutoTokenizer, AutoModelForCausalLM
+from peft import PeftModel, PeftConfig
 
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
-model_name = "BeastGokul/Bio-Mistral-7B-finetuned"
-model = AutoModelForCausalLM.from_pretrained(model_name)
-tokenizer = AutoTokenizer.from_pretrained(model_name)
+config = PeftConfig.from_pretrained("BeastGokul/Bio-Mistral-7B-finetuned")
+base_model = AutoModelForCausalLM.from_pretrained("BioMistral/BioMistral-7B")
+model = PeftModel.from_pretrained(base_model, "BeastGokul/Bio-Mistral-7B-finetuned")