elucidator8918 commited on
Commit
8d36a0a
·
verified ·
1 Parent(s): 352251d

Update tasks/text.py

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