Drazcat-AI commited on
Commit
dd15d15
·
verified ·
1 Parent(s): a0a87be

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +0 -18
handler.py CHANGED
@@ -346,24 +346,6 @@ def process_image_with_bboxes(self, image_url, picture_id, visit_id, minimal_acc
346
  class EndpointHandler():
347
  def __init__(self, path=""):
348
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
349
- """
350
- model_filename_detergentes = "model_curriculum4/prototypical_model_best_detergentes.pth"
351
- local_model_path_detergentes = hf_hub_download(repo_id="Drazcat-AI/redes_prototipicas", filename=model_filename_detergentes)
352
- self.encoder_detergentes, self.class_names_detergentes, self.prototypes_detergentes, self.eval_transform_detergentes = load_classification_model_optimized(
353
- local_model_path_detergentes, device)
354
- model_filename_mascotas = "model_curriculum4/prototypical_model_best_mascotas.pth"
355
- local_model_path_mascotas = hf_hub_download(repo_id="Drazcat-AI/redes_prototipicas", filename=model_filename_mascotas)
356
- self.encoder_mascotas, self.class_names_mascotas, self.prototypes_mascotas, self.eval_transform_mascotas = load_classification_model_optimized(
357
- local_model_path_mascotas, device)
358
- model_filename_vinos = "model_curriculum4/prototypical_model_best_vinos.pth"
359
- local_model_path_vinos = hf_hub_download(repo_id="Drazcat-AI/redes_prototipicas", filename=model_filename_vinos)
360
- self.encoder_vinos, self.class_names_vinos, self.prototypes_vinos, self.eval_transform_vinos = load_classification_model_optimized(
361
- local_model_path_vinos, device)
362
- model_filename_cecinas = "model_curriculum4/prototypical_model_best_cecinas.pth"
363
- local_model_path_cecinas = hf_hub_download(repo_id="Drazcat-AI/redes_prototipicas", filename=model_filename_cecinas)
364
- self.encoder_cecinas, self.class_names_cecinas, self.prototypes_cecinas, self.eval_transform_cecinas = load_classification_model_optimized(
365
- local_model_path_cecinas, device)
366
- """
367
  model_filename_licores = "model_curriculum4/prototypical_model_best_licores.pth"
368
  local_model_path_licores = hf_hub_download(repo_id="Drazcat-AI/redes_prototipicas", filename=model_filename_licores)
369
  self.encoder_licores, self.class_names_licores, self.prototypes_licores, self.eval_transform_licores = load_classification_model_optimized(
 
346
  class EndpointHandler():
347
  def __init__(self, path=""):
348
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
349
  model_filename_licores = "model_curriculum4/prototypical_model_best_licores.pth"
350
  local_model_path_licores = hf_hub_download(repo_id="Drazcat-AI/redes_prototipicas", filename=model_filename_licores)
351
  self.encoder_licores, self.class_names_licores, self.prototypes_licores, self.eval_transform_licores = load_classification_model_optimized(