Spaces:
Sleeping
Sleeping
Upload app.py
Browse files
app.py
CHANGED
@@ -36,6 +36,10 @@ def predict(img):
|
|
36 |
img = cv2.resize(img, (img_size, img_size))
|
37 |
img = img.reshape(1, img_size, img_size, 1)
|
38 |
preds = model.predict(img)[0]
|
|
|
|
|
|
|
|
|
39 |
return {label: float(pred) for label, pred in zip(labels, preds)}
|
40 |
except Exception as e:
|
41 |
# Print the exception to the console
|
|
|
36 |
img = cv2.resize(img, (img_size, img_size))
|
37 |
img = img.reshape(1, img_size, img_size, 1)
|
38 |
preds = model.predict(img)[0]
|
39 |
+
|
40 |
+
# Print the predictions
|
41 |
+
print("Predictions:", preds)
|
42 |
+
|
43 |
return {label: float(pred) for label, pred in zip(labels, preds)}
|
44 |
except Exception as e:
|
45 |
# Print the exception to the console
|