Update app.py

app.py

@@ -36,7 +36,7 @@ def predict_top_classes(img, model_name):
     temp = model.predict(x)
 
     idx = np.argsort(np.squeeze(temp))[::-1]
-    top5_value = np.asarray([temp[0][i] for i in idx[0:5]])
+    top5_value = np.asarray([temp[0][i] for i in idx[0:5])
     top5_idx = idx[0:5]
 
     return {CLASS_LABEL[i]: str(v) for i, v in zip(top5_idx, top5_value)}
@@ -48,16 +48,22 @@ models = {
     "DenseNet201": 4
 }
 
+def dropdown_example(choice, img):
+    model_name = models[choice]
+    return predict_top_classes(img, model_name)
+
+dropdown = gr.inputs.Dropdown(
+    choices=list(models.keys()),
+    type="str",
+    label="Select a model"
+)
+
+image_input = gr.inputs.Image(type='pil')
+
 interface = gr.Interface(
-    predict_top_classes,
-    [
-        gr.inputs.Image(type='pil'),
-        gr.Dropdown(
-            choices=list(models.keys()),
-            value=1,
-            label="Select a model"
-        )
-    ],
+    fn=dropdown_example,
+    inputs=[dropdown, image_input],
     outputs='label'
 )
-interface.launch()
+
+interface.launch()