Spaces:

phamson02
/

tho_ai

Runtime error

phamson02 commited on Jun 29, 2023

Commit

cd29b18

1 Parent(s): 0eb9251

update

Files changed (1) hide show

app.py CHANGED Viewed

@@ -28,7 +28,7 @@ def generate_poem(text, style):
     model = models[style]
     # Tokenize the input line
-    input_ids = tokenizer.encode(text, return_tensors="pt")
     # Generate text
     output = model.generate(input_ids, max_length=100, do_sample=True, temperature=0.7)

     model = models[style]
     # Tokenize the input line
+    input_ids = tokenizer.encode(text, return_tensors="pt")[:, :-1]
     # Generate text
     output = model.generate(input_ids, max_length=100, do_sample=True, temperature=0.7)