Fix output image sizes
Browse files
app.py
CHANGED
|
@@ -113,5 +113,5 @@ def _load_model(model_name: str):
|
|
| 113 |
model_finetuned.eval()
|
| 114 |
|
| 115 |
_load_model('ar_base.tar')
|
| 116 |
-
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=(
|
| 117 |
demo.launch(debug=True)
|
|
|
|
| 113 |
model_finetuned.eval()
|
| 114 |
|
| 115 |
_load_model('ar_base.tar')
|
| 116 |
+
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}')
|
| 117 |
demo.launch(debug=True)
|