Spaces:

omkar56
/

text_generation_model

Sleeping

omkar56 commited on Dec 6, 2023

Commit

58ef054

1 Parent(s): cf9f7eb

Update main.py

Files changed (1) hide show

main.py CHANGED Viewed

@@ -38,4 +38,4 @@ async def generate_text(request: Request, prompt: str = Body()):
         seed=random.randint(0, 10**7),
     )[0]
-    return response.token.text

         seed=random.randint(0, 10**7),
     )[0]
+    return response