Spaces:
Sleeping
Sleeping
removed comments
Browse files- tasks/text.py +0 -6
tasks/text.py
CHANGED
@@ -74,9 +74,6 @@ async def evaluate_text(request: TextEvaluationRequest):
|
|
74 |
# Make random predictions (placeholder for actual model inference)
|
75 |
true_labels = test_dataset["label"]
|
76 |
|
77 |
-
# load the xgboost model
|
78 |
-
# with open("models/stella_400_xgb_500.pkl",'rb') as f:
|
79 |
-
# xgbclassifier = pickle.load(f)
|
80 |
from torch import nn, optim
|
81 |
|
82 |
class SimpleNN2(nn.Module):
|
@@ -103,9 +100,6 @@ async def evaluate_text(request: TextEvaluationRequest):
|
|
103 |
# Set the model to evaluation mode
|
104 |
model_nn.eval()
|
105 |
|
106 |
-
# make inference
|
107 |
-
# predictions = xgbclassifier.predict(embeddings)
|
108 |
-
|
109 |
# Make predictions
|
110 |
with torch.no_grad():
|
111 |
outputs = model_nn(embeddings)
|
|
|
74 |
# Make random predictions (placeholder for actual model inference)
|
75 |
true_labels = test_dataset["label"]
|
76 |
|
|
|
|
|
|
|
77 |
from torch import nn, optim
|
78 |
|
79 |
class SimpleNN2(nn.Module):
|
|
|
100 |
# Set the model to evaluation mode
|
101 |
model_nn.eval()
|
102 |
|
|
|
|
|
|
|
103 |
# Make predictions
|
104 |
with torch.no_grad():
|
105 |
outputs = model_nn(embeddings)
|