Update handler.py
Browse files- handler.py +1 -0
handler.py
CHANGED
|
@@ -81,6 +81,7 @@ class EndpointHandler():
|
|
| 81 |
model_path = hf_hub_download(repo_id="Drazcat-AI/categories_peru", filename=model_filename)
|
| 82 |
class_path = hf_hub_download(repo_id="Drazcat-AI/categories_peru", filename='vit_multiclass_model_best/class_info.json')
|
| 83 |
config_path = hf_hub_download(repo_id="Drazcat-AI/categories_peru", filename='vit_multiclass_model_best/processor_config.json')
|
|
|
|
| 84 |
self.model, self.image_processor, self.class_info, self.device = load_model_and_config(model_path, class_path, config_path)
|
| 85 |
|
| 86 |
def predict_objects(self, image_url, accuracy):
|
|
|
|
| 81 |
model_path = hf_hub_download(repo_id="Drazcat-AI/categories_peru", filename=model_filename)
|
| 82 |
class_path = hf_hub_download(repo_id="Drazcat-AI/categories_peru", filename='vit_multiclass_model_best/class_info.json')
|
| 83 |
config_path = hf_hub_download(repo_id="Drazcat-AI/categories_peru", filename='vit_multiclass_model_best/processor_config.json')
|
| 84 |
+
config_path = hf_hub_download(repo_id="Drazcat-AI/categories_peru", filename='vit_multiclass_model_best/config.json')
|
| 85 |
self.model, self.image_processor, self.class_info, self.device = load_model_and_config(model_path, class_path, config_path)
|
| 86 |
|
| 87 |
def predict_objects(self, image_url, accuracy):
|