Tonic commited on
Commit
fb145a1
1 Parent(s): b01335d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import gradio as gr
5
  base_model_id = "mistralai/Mistral-7B-v0.1"
6
 
7
  # Load the fine-tuned model "Tonic/mistralmed"
8
- model = AutoModelForCausalLM.from_pretrained("Tonic/mistralmed")
9
 
10
  tokenizer = AutoTokenizer.from_pretrained(base_model_id, trust_remote_code=True)
11
  tokenizer.pad_token = tokenizer.eos_token
 
5
  base_model_id = "mistralai/Mistral-7B-v0.1"
6
 
7
  # Load the fine-tuned model "Tonic/mistralmed"
8
+ model = AutoModelForCausalLM.from_pretrained("Tonic/mistralmed", token="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")
9
 
10
  tokenizer = AutoTokenizer.from_pretrained(base_model_id, trust_remote_code=True)
11
  tokenizer.pad_token = tokenizer.eos_token