Update app.py

app.py

@@ -14,7 +14,7 @@ def generate_story(text):
     prompt = PromptTemplate(
         input_variables=["text"],
         template=""" 
-         You are a fun and seasoned storyteller. Generate a story for me about {text}.
+         You are a fun and seasoned storyteller. Generate a short story for a 5 years old audience about {text}.
                  """
     )
     story = LLMChain(llm=llm, prompt=prompt)
@@ -24,4 +24,12 @@ def app(text):
     story = generate_story(text)
     return story
 
-gr.Interface(fn=app, inputs=[gr.Textbox()], outputs=[gr.Textbox()]).launch()
+with gr.Blocks() as demo:
+    with gr.Column():
+        text = gr.Textbox()
+        submit_btn = gr.Button('Submit')
+        story = gr.Textbox()
+
+    submit_btn.click(fn=app, inputs=[text], outputs=[story])
+        
+demo.launch()