Spaces:
Sleeping
Sleeping
Commit
·
1b17bcf
1
Parent(s):
e81eb6e
updated app.py file
Browse files
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="
|
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 |
|