Spaces:
Paused
Paused
Update app/main.py
Browse files- 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
|