Hugging Face's logo

Drazcat-AI
/
categories_peru

Image Classification
Transformers
Safetensors
PyTorch
English
computer-vision
Model card Files
xet
 Community
Drazcat-AI commited on
Commit
d25d00f
·
verified ·
1 Parent(s): 32cf20a

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +2 -2
handler.py CHANGED
@@ -83,8 +83,8 @@ class EndpointHandler():
83
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
84
  model_filename = "vit_multiclass_model_best/model.safetensors"
85
  local_path = hf_hub_download(repo_id="Drazcat-AI/categories_peru", filename=model_filename)
86
- hf_hub_download(repo_id="Drazcat-AI/categories_peru", filename='class_info.json')
87
- hf_hub_download(repo_id="Drazcat-AI/categories_peru", filename='processor_config.json')
88
  print('MODEL PATH:',local_path)
89
  self.model, self.image_processor, self.class_info, self.device = load_model_and_config(local_path)
90
 
 
83
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
84
  model_filename = "vit_multiclass_model_best/model.safetensors"
85
  local_path = hf_hub_download(repo_id="Drazcat-AI/categories_peru", filename=model_filename)
86
+ hf_hub_download(repo_id="Drazcat-AI/categories_peru", filename='vit_multiclass_model_best/class_info.json')
87
+ hf_hub_download(repo_id="Drazcat-AI/categories_peru", filename='vit_multiclass_model_best/processor_config.json')
88
  print('MODEL PATH:',local_path)
89
  self.model, self.image_processor, self.class_info, self.device = load_model_and_config(local_path)
90