Upload test_model.py with huggingface_hub
Browse files- test_model.py +19 -0
test_model.py
ADDED
|
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
"""
|
| 2 |
+
Simple test script for the computer vision model
|
| 3 |
+
"""
|
| 4 |
+
|
| 5 |
+
from image_classifier import ImageClassifier
|
| 6 |
+
|
| 7 |
+
def test_model():
|
| 8 |
+
print("Testing Computer Vision Model:")
|
| 9 |
+
print("-" * 40)
|
| 10 |
+
|
| 11 |
+
classifier = ImageClassifier()
|
| 12 |
+
print("Image classifier initialized successfully!")
|
| 13 |
+
print("Model can classify images into 1000 different categories")
|
| 14 |
+
print("To test with an image, use: classifier.classify_image('path/to/image.jpg')")
|
| 15 |
+
print()
|
| 16 |
+
print("Note: For actual testing, you'll need to provide a valid image file.")
|
| 17 |
+
|
| 18 |
+
if __name__ == "__main__":
|
| 19 |
+
test_model()
|