Hugging Face's logo

Spaces:
pseudolab
/
MistralMED_Chat
Paused

App Files Community
9
Tonic commited on
Commit
5ea0370
•
1 Parent(s): abe73b7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ tokenizer.padding_side = 'left'
14
 
15
  # Load the PEFT model
16
  peft_config = PeftConfig.from_pretrained("Tonic/mistralmed", token="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")
17
- model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-v0.1")
18
  peft_model = PeftModel.from_pretrained(base_model, "Tonic/mistralmed", token="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")
19
 
20
  class ChatBot:
 
14
 
15
  # Load the PEFT model
16
  peft_config = PeftConfig.from_pretrained("Tonic/mistralmed", token="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")
17
+ peft_model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-v0.1")
18
  peft_model = PeftModel.from_pretrained(base_model, "Tonic/mistralmed", token="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")
19
 
20
  class ChatBot: