computer-vision-demo / test_model.py
Ahmed766's picture
Upload test_model.py with huggingface_hub
f797d36 verified
"""
Simple test script for the computer vision model
"""
from image_classifier import ImageClassifier
def test_model():
print("Testing Computer Vision Model:")
print("-" * 40)
classifier = ImageClassifier()
print("Image classifier initialized successfully!")
print("Model can classify images into 1000 different categories")
print("To test with an image, use: classifier.classify_image('path/to/image.jpg')")
print()
print("Note: For actual testing, you'll need to provide a valid image file.")
if __name__ == "__main__":
test_model()