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)
|