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 = ''
|