asoria HF staff commited on
Commit
638b67a
·
verified ·
1 Parent(s): 7a382dd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -52,7 +52,7 @@ def query_remote_model(text):
52
  return pred[0]["generated_text"]
53
 
54
  def query_local_model_transformers(text):
55
- pred = pipe(text)
56
  print(type(pred))
57
  print(pred)
58
  return pred
 
52
  return pred[0]["generated_text"]
53
 
54
  def query_local_model_transformers(text):
55
+ pred = pipe(text, max_length=1000)
56
  print(type(pred))
57
  print(pred)
58
  return pred