Update app.py

app.py

@@ -59,7 +59,7 @@ tokenizer.padding_side = 'left'
 
 # Load the PEFT model
 peft_config = PeftConfig.from_pretrained("Tonic/stablemed", token=hf_token)
-peft_model = MistralForCausalLM.from_pretrained("stabilityai/stablelm-3b-4e1t", token=hf_token, trust_remote_code=True)
+peft_model = AutoModelForCausalLM.from_pretrained("stabilityai/stablelm-3b-4e1t", token=hf_token, trust_remote_code=True)
 peft_model = PeftModel.from_pretrained(peft_model, "Tonic/stablemed", token=hf_token)
 
 class ChatBot: