Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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,
|
| 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 = ''
|