Drazcat-AI
/

categories_peru

Image Classification

computer-vision

Model card Files Files and versions

Drazcat-AI commited on Jul 14, 2025

Commit

bd025b7

·

verified ·

1 Parent(s): 6632aa8

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -26,7 +26,7 @@ def load_model_and_config(model_path, class_path, config_path):
 	)
 	# Cargar modelo
-	model_path = model_path.replace('model.safetensors')
 	model = ViTForImageClassification.from_pretrained(model_path)
 	model.eval()

 	)
 	# Cargar modelo
+	model_path = model_path.replace('model.safetensors', '')
 	model = ViTForImageClassification.from_pretrained(model_path)
 	model.eval()