Commit ·
fea8f7d
1
Parent(s): 4f8c74b
Your descriptive commit message
Browse files
app.py
CHANGED
|
@@ -5,7 +5,7 @@ from PIL import Image
|
|
| 5 |
|
| 6 |
|
| 7 |
cwd = os.getcwd()
|
| 8 |
-
model_path = os.path.join(cwd,'
|
| 9 |
classifier = ImageClassifier(model_path=model_path)
|
| 10 |
|
| 11 |
def classify_image(image):
|
|
|
|
| 5 |
|
| 6 |
|
| 7 |
cwd = os.getcwd()
|
| 8 |
+
model_path = os.path.join(cwd,'cnn_128_model_100epochs.pth')
|
| 9 |
classifier = ImageClassifier(model_path=model_path)
|
| 10 |
|
| 11 |
def classify_image(image):
|