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)
|