Matej commited on
Commit
adf8c13
1 Parent(s): b8fb8cf
Files changed (1) hide show
  1. my_app.py +1 -0
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())