Spaces:
Sleeping
Sleeping
Update tasks/text.py
Browse files- tasks/text.py +2 -1
tasks/text.py
CHANGED
@@ -321,7 +321,8 @@ async def evaluate_text(request: TextEvaluationRequest):
|
|
321 |
# YOUR MODEL INFERENCE CODE HERE
|
322 |
# Update the code below to replace the random baseline by your model inference within the inference pass where the energy consumption and emissions are tracked.
|
323 |
#--------------------------------------------------------------------------------------------
|
324 |
-
|
|
|
325 |
model.eval()
|
326 |
for batch in tqdm(test_dataloader):
|
327 |
batch = tuple(t.to(device) for t in batch)
|
|
|
321 |
# YOUR MODEL INFERENCE CODE HERE
|
322 |
# Update the code below to replace the random baseline by your model inference within the inference pass where the energy consumption and emissions are tracked.
|
323 |
#--------------------------------------------------------------------------------------------
|
324 |
+
|
325 |
+
predictions = []
|
326 |
model.eval()
|
327 |
for batch in tqdm(test_dataloader):
|
328 |
batch = tuple(t.to(device) for t in batch)
|