harishnair04 commited on
Commit
8ed0a63
1 Parent(s): 0254726

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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
- # tokenizer.pad_token_id = tokenizer.eos_token_id
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}"