yuanwenyue commited on
Commit
0ae79fe
1 Parent(s): c1ff23a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -214,7 +214,7 @@ if __name__ == "__main__":
214
  options = ['DINOv2', 'DINOv2-reg', 'CLIP', 'MAE', 'DeiT-III']
215
  model_option = gr.Radio(options, value="DINOv2", label='Choose a 2D foundation model')
216
  kmeans_num = gr.Number(
217
- label="number of K-Means clusters", value=20
218
  )
219
 
220
  timm_model_card = {
@@ -287,11 +287,11 @@ if __name__ == "__main__":
287
  inputs=[image_input, model_option, kmeans_num],
288
  outputs="plot",
289
  examples=[
290
- ["/tmp/examples/library.jpg", "DINOv2"],
291
- ["/tmp/examples/livingroom.jpg", "DINOv2"],
292
- ["/tmp/examples/airplane.jpg", "DINOv2"],
293
- ["/tmp/examples/ship.jpg", "DINOv2"],
294
- ["/tmp/examples/chair.jpg", "DINOv2"],
295
  ])
296
  demo.launch()
297
 
 
214
  options = ['DINOv2', 'DINOv2-reg', 'CLIP', 'MAE', 'DeiT-III']
215
  model_option = gr.Radio(options, value="DINOv2", label='Choose a 2D foundation model')
216
  kmeans_num = gr.Number(
217
+ label="Number of K-Means clusters", value=20
218
  )
219
 
220
  timm_model_card = {
 
287
  inputs=[image_input, model_option, kmeans_num],
288
  outputs="plot",
289
  examples=[
290
+ ["/tmp/examples/library.jpg", "DINOv2", 20],
291
+ ["/tmp/examples/livingroom.jpg", "DINOv2", 20],
292
+ ["/tmp/examples/airplane.jpg", "DINOv2", 20],
293
+ ["/tmp/examples/ship.jpg", "DINOv2", 20],
294
+ ["/tmp/examples/chair.jpg", "DINOv2", 20],
295
  ])
296
  demo.launch()
297