add error handler

app.py

@@ -22,6 +22,10 @@ class_names = {
 }
 
 def classify_image(image_file, label_file):
+    if image_file is None:
+        raise gr.Error("Need a histology image")
+    if label_file is None:
+        raise gr.Error("Need a label image")
     data = {"image":image_file, "label":label_file}
     batch = preprocess(data)
     batch['image'] = batch['image']
@@ -66,8 +70,8 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue=page_utils.KALBE_THEME_COLOR,
     with gr.Row():
         with gr.Column():
             with gr.Row():
-                inp_img = gr.Image(type="filepath", image_mode="RGB")
-                label_img = gr.Image(type="filepath", image_mode="L")
+                inp_img = gr.Image(type="filepath", image_mode="RGB", label="Histology Image", show_label=True)
+                label_img = gr.Image(type="filepath", image_mode="L", label="Label Image", show_label=True)
             with gr.Row():
                 clear_btn = gr.Button(value="Clear")
                 process_btn = gr.Button(value="Process", variant="primary")

index.html

@@ -34,8 +34,8 @@
 						</p>
 						<ul>
 							<li>Target: Nuclei</li>
-							<li>Task: Nuclei Cells Class Classification</li>
-							<li>Modality: Image (RGB)</li>
+							<li>Task: Classification</li>
+							<li>Modality: Images (Histology and Label) </li>
 						</ul>
 					</div>
 				</div>