Update handler.py
Browse files- handler.py +1 -1
handler.py
CHANGED
|
@@ -367,7 +367,7 @@ def process_image_with_bboxes(self, image_url, picture_id, visit_id, minimal_acc
|
|
| 367 |
return results_df
|
| 368 |
|
| 369 |
class EndpointHandler():
|
| 370 |
-
def __init__(self):
|
| 371 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
| 372 |
model_path_detergentes = "model_curriculum4/prototypical_model_best_detergentes.pth"
|
| 373 |
train_path_detergentes = "datasets/detergentes/train"
|
|
|
|
| 367 |
return results_df
|
| 368 |
|
| 369 |
class EndpointHandler():
|
| 370 |
+
def __init__(self, path=""):
|
| 371 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
| 372 |
model_path_detergentes = "model_curriculum4/prototypical_model_best_detergentes.pth"
|
| 373 |
train_path_detergentes = "datasets/detergentes/train"
|