English
yeswondwerr commited on
Commit
6cd45a2
1 Parent(s): 8e9df48

Update __main__.py

Browse files
Files changed (1) hide show
  1. __main__.py +1 -1
__main__.py CHANGED
@@ -198,7 +198,7 @@ def answer_question(
198
  new_text = (
199
  generated_text
200
  + "<|start_header_id|>user<|end_header_id|>\n\n"
201
- + inp
202
  + "<|start_header_id|>assistant<|end_header_id|>\n\n"
203
  )
204
  new_input_ids = tokenizer(new_text, return_tensors="pt").input_ids.to(
 
198
  new_text = (
199
  generated_text
200
  + "<|start_header_id|>user<|end_header_id|>\n\n"
201
+ + q
202
  + "<|start_header_id|>assistant<|end_header_id|>\n\n"
203
  )
204
  new_input_ids = tokenizer(new_text, return_tensors="pt").input_ids.to(