Update app.py

app.py

@@ -23,19 +23,30 @@ def generate_text(prompt, max_length=100, min_length=20, temperature=1.0):
 
     return generated_text
 
-# Create the Gradio interface
-iface = gr.Interface(
-    generate_text,
-    [
-        gr.Textbox(lines=5, label="Prompt"),
-        gr.Slider(0, 2048, 100, label="Max Length"),
-        gr.Slider(0, 2048, 20, label="Min Length"),
-        gr.Slider(0.1, 2.0, 1.0, label="Temperature"),
-    ],
-    "textbox",
-    title="TinyLlama Text Generator",
-    live=True,  # Enable live updates as the user types
-)
-
-# Launch the Gradio app
-iface.launch()
+with gr.Blocks() as demo:
+
+    prompt_txt = gr.Textbox(label="Prompt", lines=2)
+    max_len_slider = gr.Slider(0, 2048, 100, label="Max Length")
+    min_len_slider = gr.Slider(0, 2048, 20, label="Min Length")
+    temp_slider = gr.Slider(0.1, 2.0, 1.0, label="Temperature")
+
+    output_txt = gr.Textbox(value="", label="Generated Text")
+
+    submit_btn = gr.Button(value="Generate Text")
+    submit_btn.click(
+        generate_text,
+        inputs=[prompt_txt, max_len_slider, min_len_slider, temp_slider],
+        outputs=[output_txt]
+    )
+
+    gr.Markdown("## Text Generation Example")
+    gr.Examples(
+        examples=[["Write a creative story", 100, 20, 1.0]],
+        inputs=[prompt_txt, max_len_slider, min_len_slider, temp_slider],
+        outputs=output_txt,
+        fn=generate_text,
+        cache_examples=True,
+    )
+
+if __name__ == "__main__":
+    demo.launch()