app update
Browse files
app.py
CHANGED
@@ -212,14 +212,14 @@ def generate_wikipedia_outputs(prompt=None, max_new_tokens=2000):
|
|
212 |
|
213 |
title = "Nano GPT"
|
214 |
|
215 |
-
description1 = "Nano GPT trained on Shakespeare dataset
|
216 |
|
217 |
shakespeare_interface = gr.Interface(generate_shakespeare_outputs,
|
218 |
inputs=[gr.Textbox(label="Enter any prompt ", type="text", value="Once upon a time,"),
|
219 |
gr.Slider(minimum=100, maximum=5000, step=100, value=2000, label="Max new tokens")],
|
220 |
outputs=gr.Textbox(label="Output generated", type="text"), description=description1)
|
221 |
|
222 |
-
description2 = "Nano GPT trained on Wikipedia dataset
|
223 |
|
224 |
wiki_interface = gr.Interface(generate_wikipedia_outputs,
|
225 |
inputs=[gr.Textbox(label="Enter any prompt ", type="text", value="James Bond"),
|
|
|
212 |
|
213 |
title = "Nano GPT"
|
214 |
|
215 |
+
description1 = "Nano GPT trained on <a href='https://www.kaggle.com/datasets/mikeortman/wikipedia-sentences'>Shakespeare dataset</a>. It is trained on a very small amount of data to understand how GPT's are trained and built. The implementation can be found <a href='https://github.com/karpathy/nanoGPT'>here.</a>"
|
216 |
|
217 |
shakespeare_interface = gr.Interface(generate_shakespeare_outputs,
|
218 |
inputs=[gr.Textbox(label="Enter any prompt ", type="text", value="Once upon a time,"),
|
219 |
gr.Slider(minimum=100, maximum=5000, step=100, value=2000, label="Max new tokens")],
|
220 |
outputs=gr.Textbox(label="Output generated", type="text"), description=description1)
|
221 |
|
222 |
+
description2 = "Nano GPT trained on <a href='https://github.com/karpathy/char-rnn/blob/6f9487a6fe5b420b7ca9afb0d7c078e37c1d1b4e/data/tinyshakespeare/input.txt'>Wikipedia dataset</a>. It is trained on a very small amount of data to understand how GPT's are trained and built. The implementation can be found <a href='https://github.com/karpathy/nanoGPT'>here.</a>"
|
223 |
|
224 |
wiki_interface = gr.Interface(generate_wikipedia_outputs,
|
225 |
inputs=[gr.Textbox(label="Enter any prompt ", type="text", value="James Bond"),
|