Hugging Face's logo

Spaces:
0-ma
/
geometric-shape-detector
Running

App Files Community
0-ma commited on
Commit
f6afbce
•
1 Parent(s): 76f81a0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -5,15 +5,15 @@ from transformers import AutoImageProcessor, AutoModelForImageClassification
5
  import os
6
 
7
  model_names = [
 
 
 
8
  "0-ma/swin-geometric-shapes-tiny",
9
  "0-ma/mobilenet-v2-geometric-shapes",
10
  "0-ma/focalnet-geometric-shapes-tiny",
11
  "0-ma/efficientnet-b2-geometric-shapes",
12
- "0-ma/beit-geometric-shapes-base",
13
  "0-ma/mit-b0-geometric-shapes",
14
- "0-ma/vit-geometric-shapes-base",
15
  "0-ma/resnet-geometric-shapes",
16
- "0-ma/vit-geometric-shapes-tiny",
17
  ]
18
  labels = [
19
  'None',
 
5
  import os
6
 
7
  model_names = [
8
+ "0-ma/beit-geometric-shapes-base",
9
+ "0-ma/vit-geometric-shapes-tiny",
10
+ "0-ma/vit-geometric-shapes-base",
11
  "0-ma/swin-geometric-shapes-tiny",
12
  "0-ma/mobilenet-v2-geometric-shapes",
13
  "0-ma/focalnet-geometric-shapes-tiny",
14
  "0-ma/efficientnet-b2-geometric-shapes",
 
15
  "0-ma/mit-b0-geometric-shapes",
 
16
  "0-ma/resnet-geometric-shapes",
 
17
  ]
18
  labels = [
19
  'None',