Spaces:

dipankardas011
/

gpt2-bot

Runtime error

dipankardas011 commited on Jun 3, 2023

Commit

8450a77

1 Parent(s): 1acab3f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,4 +17,4 @@ async def docs_redirect():
 def generate(text: str):
     output: list[dict] = generator(text, max_length=200, num_return_sequences=1)
     print(output)
-    return output[0]

 def generate(text: str):
     output: list[dict] = generator(text, max_length=200, num_return_sequences=1)
     print(output)
+    return output[0]["generated_text"]