Drazcat-AI commited on
Commit
e174331
·
verified ·
1 Parent(s): 2e69037

Update handler.py

Browse files
Files changed (1) hide show
  1. 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"