Update app.py

app.py

@@ -18,7 +18,7 @@ vgg16_model = VGG16Classifier()
 inceptionV3_model = InceptionV3Classifier()
 mobilenet_model = MobileNetClassifier()
 
-def make_prediction(image, model_type):
+def make_prediction(image, model_type="CNN (Custom)"):
   if "CNN (Custom)" == model_type:
     top_classes, top_probs = custom_model.classify_image(image, top_k=3)
     return {CLASS_NAMES[cls_id]:str(prob) for cls_id, prob in zip(top_classes, top_probs)}
@@ -104,17 +104,18 @@ if __name__ == "__main__":
         output_label = gr.Label()
 
     gr.Markdown("## Sample Images")
-    print(os.path.join(os.path.dirname(__file__), "assets/dog_2.jpg"))
     gr.Examples(
         examples=[os.path.join(os.path.dirname(__file__), "assets/dog_2.jpg"),
                   os.path.join(os.path.dirname(__file__), "assets/truck.jpg"),
                   os.path.join(os.path.dirname(__file__), "assets/car.jpg")
                  ],
-        inputs=[img_input, model_type],
-        outputs=[output_label],
+        inputs=img_input,
+        outputs=output_label,
         fn=make_prediction,
         cache_examples=True,
     )
+
+            
     
     # app logic
     predict_btn_1.click(make_prediction, inputs=[img_input, model_type], outputs=[output_label])