Spaces:
Sleeping
Sleeping
Update tasks/text.py
Browse files- tasks/text.py +1 -1
tasks/text.py
CHANGED
@@ -53,7 +53,7 @@ async def evaluate_text(request: TextEvaluationRequest):
|
|
53 |
model = AutoModelForSequenceClassification.from_pretrained(model_repo)
|
54 |
tokenizer = AutoTokenizer.from_pretrained(model_repo)
|
55 |
|
56 |
-
device = torch.device("cuda") if torch.cuda.is_available() torch.device("cpu")
|
57 |
model = model.to(device)
|
58 |
model.eval()
|
59 |
|
|
|
53 |
model = AutoModelForSequenceClassification.from_pretrained(model_repo)
|
54 |
tokenizer = AutoTokenizer.from_pretrained(model_repo)
|
55 |
|
56 |
+
device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
|
57 |
model = model.to(device)
|
58 |
model.eval()
|
59 |
|