caarleexx commited on
Commit
f00ca36
·
verified ·
1 Parent(s): 9942547

Update app/main.py

Browse files
Files changed (1) hide show
  1. app/main.py +1 -1
app/main.py CHANGED
@@ -35,7 +35,7 @@ def load_model():
35
  model.eval()
36
  _pipe = pipeline(
37
  #"text2text-generation",
38
- "text-generation"
39
  model = model,
40
  tokenizer = _tokenizer,
41
  device = -1, # força CPU
 
35
  model.eval()
36
  _pipe = pipeline(
37
  #"text2text-generation",
38
+ "text-generation",
39
  model = model,
40
  tokenizer = _tokenizer,
41
  device = -1, # força CPU