Update app.py

app.py

@@ -48,9 +48,8 @@ def multimodal_prompt(user_input, system_prompt="You are an expert medical analy
 tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-v0.1", trust_remote_code=True, padding_side="left")
 tokenizer.pad_token = tokenizer.eos_token
 tokenizer.padding_side = 'left'
-
 peft_config = PeftConfig.from_pretrained("Tonic/mistralmed")
-peft_model = MistralForCausalLM.from_pretrained("mistralai/Mistral-7B-v0.1", trust_remote_code=True)
+peft_model = MistralForCausalLM.from_pretrained("mistralai/Mistral-7B-v0.1")
 peft_model = PeftModel.from_pretrained(peft_model, "Tonic/mistralmed")
 peft_model = peft_model.to(torch.bfloat16)
 peft_model = peft_model.to(device)
@@ -63,7 +62,7 @@ class ChatBot:
     def __init__(self):
         self.history = []
 
-def predict(self, user_input, system_prompt="You are an expert medical analyst:"):
+    def predict(self, user_input, system_prompt="You are an expert medical analyst:"):
         formatted_input = f"<s>[INST]{system_prompt} {user_input}[/INST]"
         user_input_ids = tokenizer.encode(formatted_input, return_tensors="pt")
         user_input_ids = user_input_ids.to(device)