Added error handling

app.py

@@ -144,6 +144,10 @@ def predict(
     gradcam_opacity,
     num_classes,
 ):
+    if prediction_image is None:
+        raise gr.Error(
+            "Please select one of the sample image or upload an image for prediction!"
+        )
     misclassified_images = None
     if is_misclassified:
         misclassified_images = get_misclassified_images(int(misclassified_count))