tfedoseev commited on
Commit
c6459ea
·
1 Parent(s): 6a67ea5

Update app.py

Browse files

Fix a bug in the infer function

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ model = load_model()
12
 
13
  def infer(input_ids):
14
  output_sequences = model.generate(input_ids=input_ids)
15
- return tokenizer.decode(outputs[0], skip_special_tokens=True)
16
 
17
  st.title("Stupid jokes with transformers")
18
  st.write("Write question you want to see a funny answer for.")
 
12
 
13
  def infer(input_ids):
14
  output_sequences = model.generate(input_ids=input_ids)
15
+ return tokenizer.decode(output_sequences[0], skip_special_tokens=True)
16
 
17
  st.title("Stupid jokes with transformers")
18
  st.write("Write question you want to see a funny answer for.")