milyiyo commited on
Commit
c151763
1 Parent(s): 2be3696

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -51,7 +51,7 @@ model_name = 'hackathon-pln-es/poem-gen-spanish-t5-small'
51
  tokenizer = AutoTokenizer.from_pretrained(model_name)
52
  model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
53
 
54
- def make_poem(author, text, sentiment, words):
55
  num_lines=5
56
  poem = text
57
  prev_output = ''
 
51
  tokenizer = AutoTokenizer.from_pretrained(model_name)
52
  model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
53
 
54
+ def make_poem(author, sentiment, words, text):
55
  num_lines=5
56
  poem = text
57
  prev_output = ''