Update handler.py
Browse files- 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(
|