Update app.py
Browse files
app.py
CHANGED
@@ -4,7 +4,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
|
|
4 |
from peft import PeftModel, PeftConfig
|
5 |
|
6 |
client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
|
7 |
-
tokenizer =
|
8 |
base_model = AutoModelForCausalLM.from_pretrained("BioMistral/BioMistral-7B")
|
9 |
base_model.resize_token_embeddings(len(tokenizer))
|
10 |
model = PeftModel.from_pretrained(base_model, "BeastGokul/Bio-Mistral-7B-finetuned")
|
|
|
4 |
from peft import PeftModel, PeftConfig
|
5 |
|
6 |
client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
|
7 |
+
tokenizer = AutoTokenizer.from_pretrained("BeastGokul/Bio-Mistral-7B-finetuned")
|
8 |
base_model = AutoModelForCausalLM.from_pretrained("BioMistral/BioMistral-7B")
|
9 |
base_model.resize_token_embeddings(len(tokenizer))
|
10 |
model = PeftModel.from_pretrained(base_model, "BeastGokul/Bio-Mistral-7B-finetuned")
|