Commit
•
18d9da8
1
Parent(s):
fff458d
Update app.py
Browse files
app.py
CHANGED
@@ -170,9 +170,9 @@ def generate_shakespeare_outputs(prompt=None, max_new_tokens=2000):
|
|
170 |
text_output = shakespeare_model.decode(shakespeare_model.generate(context, max_new_tokens=max_new_tokens)[0].tolist())
|
171 |
return text_output
|
172 |
|
173 |
-
title = "
|
174 |
|
175 |
-
description1 = "Nano GPT trained on <a href='https://www.kaggle.com/datasets/mikeortman/wikipedia-sentences'>Shakespeare dataset</a>.
|
176 |
|
177 |
demo = gr.Interface(generate_shakespeare_outputs,
|
178 |
inputs=[gr.Textbox(label="Enter any prompt ", type="text", value="Once upon a time,"),
|
|
|
170 |
text_output = shakespeare_model.decode(shakespeare_model.generate(context, max_new_tokens=max_new_tokens)[0].tolist())
|
171 |
return text_output
|
172 |
|
173 |
+
title = "GPT using Shakespeare Data"
|
174 |
|
175 |
+
description1 = "Nano GPT trained on <a href='https://www.kaggle.com/datasets/mikeortman/wikipedia-sentences'>Shakespeare dataset</a>. The implementation can be found <a href='https://github.com/karpathy/nanoGPT'>here.</a>"
|
176 |
|
177 |
demo = gr.Interface(generate_shakespeare_outputs,
|
178 |
inputs=[gr.Textbox(label="Enter any prompt ", type="text", value="Once upon a time,"),
|