hushell commited on
Commit
bf0f567
1 Parent(s): 1aa70ca
Files changed (1) hide show
  1. app.py +2 -2
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)