Update app.py

app.py

@@ -3,9 +3,10 @@ import gradio as gr
 
 # Use the base model's ID
 base_model_id = "mistralai/Mistral-7B-v0.1"
+model_directory = "Tonic/mistralmed"
 
 # Load the fine-tuned model "Tonic/mistralmed"
-model = AutoModelForCausalLM.from_pretrained("Tonic/mistralmed", token="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")
+model = AutoModelForCausalLM.from_pretrained(model_directory, token="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")
 
 tokenizer = AutoTokenizer.from_pretrained(base_model_id, trust_remote_code=True)
 tokenizer.pad_token = tokenizer.eos_token