nnsohamnn commited on
Commit
8cbc34c
·
verified ·
1 Parent(s): a40d279

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
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()