Spaces:
Sleeping
Sleeping
Matej
commited on
Commit
•
7c3521c
1
Parent(s):
5aab43b
debug
Browse files
my_app.py
CHANGED
|
@@ -32,7 +32,6 @@ def predict(model_selection, image):
|
|
| 32 |
model = model1 if model_selection == "EfficentNetB0 Fine Tune" else model2
|
| 33 |
|
| 34 |
image = preprocess(image)
|
| 35 |
-
print("image", image)
|
| 36 |
prediction = model.predict(image)
|
| 37 |
print("model prediction", prediction)
|
| 38 |
predicted_class = classes[int(tf.argmax(prediction, axis=1))]
|
|
|
|
| 32 |
model = model1 if model_selection == "EfficentNetB0 Fine Tune" else model2
|
| 33 |
|
| 34 |
image = preprocess(image)
|
|
|
|
| 35 |
prediction = model.predict(image)
|
| 36 |
print("model prediction", prediction)
|
| 37 |
predicted_class = classes[int(tf.argmax(prediction, axis=1))]
|