Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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(
|