Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -262,6 +262,9 @@ def create_interface():
|
|
| 262 |
gr.Markdown("# 🤖 GPT-2 450M Chat")
|
| 263 |
gr.Markdown("Custom GPT-2 trained on FineWebEdu+SmolTalk (FineWebEdu(50k) + SmolTalk(2k steps))")
|
| 264 |
|
|
|
|
|
|
|
|
|
|
| 265 |
with gr.Row():
|
| 266 |
# Sidebar - Model Selection
|
| 267 |
with gr.Column(scale=1):
|
|
@@ -404,6 +407,7 @@ def create_interface():
|
|
| 404 |
|
| 405 |
return demo
|
| 406 |
|
|
|
|
| 407 |
# Launch
|
| 408 |
if __name__ == "__main__":
|
| 409 |
demo = create_interface()
|
|
|
|
| 262 |
gr.Markdown("# 🤖 GPT-2 450M Chat")
|
| 263 |
gr.Markdown("Custom GPT-2 trained on FineWebEdu+SmolTalk (FineWebEdu(50k) + SmolTalk(2k steps))")
|
| 264 |
|
| 265 |
+
# Added Hugging Face repo link
|
| 266 |
+
gr.Markdown("Repo with training details and weight [here](https://huggingface.co/nnsohamnn/gpt2-450M-fineweb)")
|
| 267 |
+
|
| 268 |
with gr.Row():
|
| 269 |
# Sidebar - Model Selection
|
| 270 |
with gr.Column(scale=1):
|
|
|
|
| 407 |
|
| 408 |
return demo
|
| 409 |
|
| 410 |
+
|
| 411 |
# Launch
|
| 412 |
if __name__ == "__main__":
|
| 413 |
demo = create_interface()
|