zmbfeng commited on
Commit
81384e9
·
verified ·
1 Parent(s): 556b209

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -14,7 +14,8 @@ def create_response(input_str):
14
  #output_str = output_raw[0]['generated_text']
15
  #output_str = output_str.replace("\n", "")
16
  #output_str = output_str.replace(input_str, "")
17
- output_str = tokenizer.decode(model.generate(**tokenizer("What are John West's hobbies?"+tokenizer.eos_token,return_tensors="pt",max_length=200))[0])
 
18
  return (output_str)
19
 
20
  demo = gr.Interface(
 
14
  #output_str = output_raw[0]['generated_text']
15
  #output_str = output_str.replace("\n", "")
16
  #output_str = output_str.replace(input_str, "")
17
+ #output_str = tokenizer.decode(model.generate(**tokenizer("What are John West's hobbies?"+tokenizer.eos_token,return_tensors="pt",max_length=200))[0])
18
+ output_str = tokenizer.decode(model.generate(**tokenizer(input_str+tokenizer.eos_token,return_tensors="pt",max_length=200))[0])
19
  return (output_str)
20
 
21
  demo = gr.Interface(