Change layout

app.py

@@ -113,5 +113,28 @@ def _load_model(model_name: str):
 	model_finetuned.eval()
 	
 _load_model('ar_base.tar')
-demo = gr.Interface(image_classifier, gr.inputs.Image(shape=(224,224)), [gr.outputs.Label(label="Our Classification", num_top_classes=3), gr.Image(label="Our Relevance",shape=(64,64)), gr.outputs.Label(label="Original Classification", num_top_classes=3), gr.Image(label="Original Relevance",shape=(64,64))],examples=["samples/augreg_base/tank.png", "samples/augreg_base/sundial.png", "samples/augreg_base/lizard.png", "samples/augreg_base/storck.png", "samples/augreg_base/hummingbird2.png", "samples/augreg_base/hummingbird.png"], capture_session=True, css=r'#2 {width:224px !important}; #4 {width:224px !important}')
-demo.launch(debug=True)
+#demo = gr.Interface(image_classifier, gr.inputs.Image(shape=(224,224)), [gr.outputs.Label(label="Our Classification", num_top_classes=3), gr.Image(label="Our Relevance",shape=(64,64)), gr.outputs.Label(label="Original Classification", num_top_classes=3), gr.Image(label="Original Relevance",shape=(64,64))],examples=["samples/augreg_base/tank.png", "samples/augreg_base/sundial.png", "samples/augreg_base/lizard.png", "samples/augreg_base/storck.png", "samples/augreg_base/hummingbird2.png", "samples/augreg_base/hummingbird.png"], capture_session=True)
+#demo.launch(debug=True)
+
+demo = gr.Blocks()
+
+with demo:
+	gr.Markdown('Select an image and then click **submit** to see the output.')
+	
+	with gr.Row():
+		inp = gr.inputs.Image(shape=(224,224))
+	
+	outs = []
+	with gr.Row():
+		out1 = gr.outputs.Label(label="Our Classification", num_top_classes=3)
+		out2 = gr.Image(label="Our Relevance",shape=(224,224))
+	
+	with gr.Row():
+		out3 = gr.outputs.Label(label="Original Classification", num_top_classes=3)
+		out4 = gr.Image(label="Original Relevance",shape=(224,224))
+	
+	btn = gr.Button('Submit')
+	btn.click(fn=image_classifier, inputs=inp, outputs=[out1, out2, out3, out4])
+
+demo.launch()
+