morenolq commited on
Commit
c33c1f6
1 Parent(s): f96fb6b

Fix variable name error

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ text2text_generator = pipeline("text-generation", model=model, tokenizer=tokeniz
9
 
10
  def predict(text):
11
  text = text.strip()
12
- out_text = text2text_generator(input_text, max_length=384,
13
  temperature=0.7,
14
  do_sample=True,
15
  eos_token_id = tokenizer.eos_token_id,
 
9
 
10
  def predict(text):
11
  text = text.strip()
12
+ out_text = text2text_generator(text, max_length=384,
13
  temperature=0.7,
14
  do_sample=True,
15
  eos_token_id = tokenizer.eos_token_id,