Update app.py

app.py

@@ -56,7 +56,7 @@ def wrapper_generate_text(start_text, max_length, temperature):
 
 # Gradio interface
 with gr.Blocks() as iface:
-    gr.Markdown("The MinGRU model is a simplified version of the traditional Gated Recurrent Unit (GRU), designed to reduce complexity and improve efficiency.")
+    gr.Markdown("The MinGRU model is a simplified version of the traditional Gated Recurrent Unit (GRU), designed to reduce complexity and improve efficiency,Trained on the [tiny-stories](https://huggingface.co/datasets/roneneldan/TinyStories?row=19)")
     gr.Markdown("To Learn more visit this [github](https://github.com/dame-cell/MinGru/tree/main)")
     with gr.Row():
         textbox = gr.Textbox(lines=3, label="Enter your prompt", value="Once upon a time")