Update app.py
Browse files
app.py
CHANGED
@@ -7,7 +7,7 @@ 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 =
|
11 |
|
12 |
def generate_response(user_query):
|
13 |
inputs = tokenizer(user_query, return_tensors="pt")
|
|
|
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 = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-Instruct-v0.3")
|
11 |
|
12 |
def generate_response(user_query):
|
13 |
inputs = tokenizer(user_query, return_tensors="pt")
|