Spaces:
Sleeping
Sleeping
update
Browse files- tasks/text.py +1 -1
tasks/text.py
CHANGED
@@ -82,7 +82,7 @@ async def evaluate_text(request: TextEvaluationRequest):
|
|
82 |
# dataloader = DataLoader(tokenized_test, batch_size=16, shuffle=False, collate_fn=data_collator)
|
83 |
|
84 |
print("Started prediction run")
|
85 |
-
tokenized_test = tokenizer(test_dataset['quote'], padding=True, truncation=True, return_tensors='pt')
|
86 |
# Model inference
|
87 |
model.eval()
|
88 |
predictions = np.array([])
|
|
|
82 |
# dataloader = DataLoader(tokenized_test, batch_size=16, shuffle=False, collate_fn=data_collator)
|
83 |
|
84 |
print("Started prediction run")
|
85 |
+
# tokenized_test = tokenizer(test_dataset['quote'], padding=True, truncation=True, return_tensors='pt')
|
86 |
# Model inference
|
87 |
model.eval()
|
88 |
predictions = np.array([])
|