Spaces:
Sleeping
Sleeping
update app.py
Browse files
app.py
CHANGED
@@ -30,7 +30,7 @@ def sentence_builder(txt, new_tokens):
|
|
30 |
txt += " My lord, I claim your gift, my due by promise"
|
31 |
|
32 |
t_loader = DataLoaderLite(B = 8, T = 1, text_input = txt)
|
33 |
-
out = infer_the_model(device, t_loader, save1_or_load0 = 0, max_length = new_tokens)
|
34 |
return out
|
35 |
|
36 |
|
|
|
30 |
txt += " My lord, I claim your gift, my due by promise"
|
31 |
|
32 |
t_loader = DataLoaderLite(B = 8, T = 1, text_input = txt)
|
33 |
+
out = infer_the_model(device, t_loader, save1_or_load0 = 0, max_length = int(new_tokens))
|
34 |
return out
|
35 |
|
36 |
|