Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
@@ -23,7 +23,7 @@ tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-v0.1", trust_rem
|
|
23 |
# Load the PEFT model
|
24 |
peft_config = PeftConfig.from_pretrained("Tonic/mistralmed", token="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")
|
25 |
peft_model = MistralForCausalLM.from_pretrained("mistralai/Mistral-7B-v0.1", trust_remote_code=True)
|
26 |
-
peft_model = PeftModel.from_pretrained(
|
27 |
|
28 |
class ChatBot:
|
29 |
def __init__(self):
|
|
|
23 |
# Load the PEFT model
|
24 |
peft_config = PeftConfig.from_pretrained("Tonic/mistralmed", token="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")
|
25 |
peft_model = MistralForCausalLM.from_pretrained("mistralai/Mistral-7B-v0.1", trust_remote_code=True)
|
26 |
+
peft_model = PeftModel.from_pretrained(peft_model, "Tonic/mistralmed", token="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")
|
27 |
|
28 |
class ChatBot:
|
29 |
def __init__(self):
|