Update app.py

app.py

@@ -8,8 +8,7 @@ def validate(item_id: str, threshold: float):
     images, domain, probas, valid_probas, invalid_probas = detector.predict_item_probas(item_id)
     valid_images = [x for i, x in enumerate(images) if valid_probas[i].squeeze() >= threshold]
     invalid_images = [x for i, x in enumerate(images) if valid_probas[i].squeeze() < threshold]
-    return valid_images, invalid_images
-
+    return f"<h2>Domain: {domain}</h2>", valid_images, invalid_images
 
 with gr.Blocks() as demo:
     with gr.Tabs():
@@ -17,11 +16,9 @@ with gr.Blocks() as demo:
             item_id = gr.Textbox(label="Item ID")
             threshold = gr.Number(label="Threshold", value=0.5)
             submit = gr.Button("Submit")
+            domain = gr.HTML()
             valid = gr.Gallery(label="Valid images").style(grid=[1, 2, 3], height="auto")
             invalid = gr.Gallery(label="Invalid images").style(grid=[1, 2, 3], height="auto")
-            submit.click(inputs=[item_id, threshold], outputs=[valid, invalid], fn=validate)
-
+            submit.click(inputs=[item_id, threshold], outputs=[domain, valid, invalid], fn=validate)
 
 demo.launch()
-            
-