Spaces:
Sleeping
Sleeping
Commit
·
1770e8b
1
Parent(s):
7f7b222
add missing colon
Browse files
app.py
CHANGED
@@ -149,7 +149,7 @@ model = torch.load('complete-model.pt')
|
|
149 |
|
150 |
# inference
|
151 |
slider_value = st.slider('Amount of text to generate', min_value=100, max_value=2000, value=500, step=5)
|
152 |
-
if st.button('Generat text')
|
153 |
context = torch.zeros((1, 1), dtype=torch.long, device='cuda')
|
154 |
text = model.generate(context, max_new_tokens=slider_value)[0].tolist()
|
155 |
st.json(decode(text))
|
|
|
149 |
|
150 |
# inference
|
151 |
slider_value = st.slider('Amount of text to generate', min_value=100, max_value=2000, value=500, step=5)
|
152 |
+
if st.button('Generat text'):
|
153 |
context = torch.zeros((1, 1), dtype=torch.long, device='cuda')
|
154 |
text = model.generate(context, max_new_tokens=slider_value)[0].tolist()
|
155 |
st.json(decode(text))
|