Spaces:
Runtime error
Runtime error
harishnair04
commited on
Commit
•
8ed0a63
1
Parent(s):
0254726
Update app.py
Browse files
app.py
CHANGED
@@ -45,7 +45,7 @@ model_id = "harishnair04/Gemma-medtr-2b-sft-v2"
|
|
45 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
46 |
gemma_model = AutoModelForCausalLM.from_pretrained(model_id)
|
47 |
|
48 |
-
|
49 |
|
50 |
def respond(input1):
|
51 |
template = "Instruction:\n{instruction}\n\nResponse:\n{response}"
|
|
|
45 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
46 |
gemma_model = AutoModelForCausalLM.from_pretrained(model_id)
|
47 |
|
48 |
+
tokenizer.pad_token_id = tokenizer.eos_token_id
|
49 |
|
50 |
def respond(input1):
|
51 |
template = "Instruction:\n{instruction}\n\nResponse:\n{response}"
|