SAA+

app.py

@@ -90,15 +90,17 @@ def func(image, anomaly_description, object_name, object_number, mask_number, ar
 
 
 with gr.Blocks() as demo:
-    image = gr.Image(label="Image")
-    anomaly_description = gr.Textbox(label="Anomaly Description")
-    object_name = gr.Textbox(label="Object Name")
-    object_number = gr.Textbox(label="Object Number")
-    mask_number = gr.Textbox(label="Mask Number")
-    area_threashold = gr.Textbox(label="Area Threshold")
-
-    anomaly_score = gr.Image(label="Anomaly Score")
-    saliency_map = gr.Image(label="Saliency Map")
+    with gr.Column():
+        with gr.Row():
+            image = gr.Image(label="Image")
+            anomaly_description = gr.Textbox(label="Anomaly Description")
+            object_name = gr.Textbox(label="Object Name")
+            object_number = gr.Textbox(label="Object Number")
+            mask_number = gr.Textbox(label="Mask Number")
+            area_threashold = gr.Textbox(label="Area Threshold")
+        with gr.Row():
+            anomaly_score = gr.Image(label="Anomaly Score")
+            saliency_map = gr.Image(label="Saliency Map")
 
     greet_btn = gr.Button("Inference")
     greet_btn.click(fn=func,