Ahmed766 commited on
Commit
f797d36
·
verified ·
1 Parent(s): b2d524f

Upload test_model.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. 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()