chbsaikiran commited on
Commit
1b17bcf
·
1 Parent(s): e81eb6e

updated app.py file

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -1,6 +1,5 @@
1
  import torch
2
  from torch import nn
3
- from tokenizers import Tokenizer
4
  from transformers import AutoTokenizer
5
  import gradio as gr
6
  from model import LlamaForCausalLM # Import your custom model class
@@ -65,7 +64,7 @@ demo = gr.Interface(
65
  gr.Slider(10, 100, value=50, label="Top-k")
66
  ],
67
  outputs=gr.Textbox(label="Generated Text", lines=5),
68
- title="🦙 Custom SmolLLM Demo",
69
  description="A 135M parameter language model trained on smollm-corpus"
70
  )
71
 
 
1
  import torch
2
  from torch import nn
 
3
  from transformers import AutoTokenizer
4
  import gradio as gr
5
  from model import LlamaForCausalLM # Import your custom model class
 
64
  gr.Slider(10, 100, value=50, label="Top-k")
65
  ],
66
  outputs=gr.Textbox(label="Generated Text", lines=5),
67
+ title="SmolLM2 Demo",
68
  description="A 135M parameter language model trained on smollm-corpus"
69
  )
70