Tonic commited on
Commit
7f45f98
1 Parent(s): 81395fc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -13,9 +13,9 @@ tokenizer.pad_token = tokenizer.eos_token
13
  tokenizer.padding_side = 'left'
14
 
15
  # Load the PEFT model
16
- peft_config = PeftConfig.from_pretrained("Tonic/mistralmed")
17
  base_model = AutoModelForSeq2SeqLM.from_pretrained(model_directory)
18
- peft_model = PeftModel.from_pretrained(base_model, "Tonic/mistralmed")
19
 
20
  class ChatBot:
21
  def __init__(self):
 
13
  tokenizer.padding_side = 'left'
14
 
15
  # Load the PEFT model
16
+ peft_config = PeftConfig.from_pretrained("Tonic/mistralmed", token="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")
17
  base_model = AutoModelForSeq2SeqLM.from_pretrained(model_directory)
18
+ peft_model = PeftModel.from_pretrained(base_model, "Tonic/mistralmed", token="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")
19
 
20
  class ChatBot:
21
  def __init__(self):