Spaces:
Build error
Build error
tweak
Browse files
app.py
CHANGED
|
@@ -114,7 +114,7 @@ def inference(query, labels, n_supp=10):
|
|
| 114 |
|
| 115 |
# vis
|
| 116 |
axs[idx, j].imshow(x_im)
|
| 117 |
-
axs[idx, j].set_title(f'{y}{j}')
|
| 118 |
axs[idx, j].axis('off')
|
| 119 |
|
| 120 |
x_im = preprocess(x_im) # (3, H, W)
|
|
@@ -153,4 +153,4 @@ gr.Interface(fn=inference,
|
|
| 153 |
gr.outputs.Label(label="Predicted class probabilities"),
|
| 154 |
gr.outputs.Image(type='plot', label="Support examples from Google image search"),
|
| 155 |
],
|
| 156 |
-
description="PMF few-shot learning with Google image search").launch()
|
|
|
|
| 114 |
|
| 115 |
# vis
|
| 116 |
axs[idx, j].imshow(x_im)
|
| 117 |
+
axs[idx, j].set_title(f'{y}{j}:{x.path}')
|
| 118 |
axs[idx, j].axis('off')
|
| 119 |
|
| 120 |
x_im = preprocess(x_im) # (3, H, W)
|
|
|
|
| 153 |
gr.outputs.Label(label="Predicted class probabilities"),
|
| 154 |
gr.outputs.Image(type='plot', label="Support examples from Google image search"),
|
| 155 |
],
|
| 156 |
+
description="PMF few-shot learning with Google image search").launch(debug=True)
|