Spaces:
Build error
Build error
Commit
•
629a599
1
Parent(s):
5006324
Update app.py
Browse files
app.py
CHANGED
@@ -17,7 +17,7 @@ def run_model(img):
|
|
17 |
|
18 |
new_image = load_image(img)
|
19 |
classes = ['1', '2', '3', '4', '5', '6', '7', '8', '9', 'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K', 'L', 'M', 'N', 'O', 'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z']
|
20 |
-
img = img.reshape((-1,
|
21 |
result = model.predict(img)
|
22 |
results = dict(zip(classes, result[0]))
|
23 |
return max(results, key = results.get)
|
|
|
17 |
|
18 |
new_image = load_image(img)
|
19 |
classes = ['1', '2', '3', '4', '5', '6', '7', '8', '9', 'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K', 'L', 'M', 'N', 'O', 'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z']
|
20 |
+
img = img.reshape((-1, 299, 299, 3))
|
21 |
result = model.predict(img)
|
22 |
results = dict(zip(classes, result[0]))
|
23 |
return max(results, key = results.get)
|