Spaces:
Sleeping
Sleeping
Matej
commited on
Commit
•
adf8c13
1
Parent(s):
b8fb8cf
debug
Browse files
my_app.py
CHANGED
@@ -32,6 +32,7 @@ def preprocess(image):
|
|
32 |
|
33 |
def predict(model_selection, image):
|
34 |
# Choose the model based on the dropdown selection
|
|
|
35 |
model = model1 if model_selection == "EfficentNetB0 Fine Tune" else model2
|
36 |
|
37 |
print(model.summary())
|
|
|
32 |
|
33 |
def predict(model_selection, image):
|
34 |
# Choose the model based on the dropdown selection
|
35 |
+
print("---model_selection---", model_selection) #
|
36 |
model = model1 if model_selection == "EfficentNetB0 Fine Tune" else model2
|
37 |
|
38 |
print(model.summary())
|