mkthoma commited on
Commit
754fe61
·
1 Parent(s): 46b2f47

app update

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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. 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 Wikipedia dataset. 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"),
 
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"),