Update app.py

app.py

@@ -14,5 +14,5 @@ class TextInput(BaseModel):
 @app.post("/generate")
 def generate_text(text_input: TextInput):
     # Generate text from the model based on the input
-    generated = generator(text_input.input_text, max_length=100, num_return_sequences=1)
-    return generated[0]['generated_text']
+    generated = generator(text_input.input_text, max_length=150, num_return_sequences=1)
+    return {"generated_text": generated[0]['generated_text']}