raja5259 commited on
Commit
c3ea996
1 Parent(s): e7c0e29

update app.py

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