Update app.py

app.py

@@ -15,4 +15,4 @@ class TextInput(BaseModel):
 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_text": generated[0]['generated_text']}
+    return generated