Spaces:

zihaoz96
/

shark-classifier

Runtime error

zihaoz96 commited on Dec 16, 2021

Commit

7ea38a3

•

1 Parent(s): 64d2c38

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,7 +18,7 @@ categories = open("categories.txt", "r")
 labels = categories.readline().split(";")
 # create a radio
-radio = gr.inputs.Radio(models_name, default="DenseNet", type="value")
 def predict_image(image, model_name):
     # model create by keras

 labels = categories.readline().split(";")
 # create a radio
+radio = gr.inputs.Radio(models_name, default="DenseNet121", type="value")
 def predict_image(image, model_name):
     # model create by keras