Hugging Face's logo

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

App Files Community
0-ma commited on
Commit
479feb3
·
verified ·
1 Parent(s): fdab764

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ labels = [
15
 
16
  # Available models for the dropdown
17
  models = {
18
- # "0-ma/swin-geometric-shapes-tiny": "0-ma/swin-geometric-shapes-tiny", TODO: Fix preprocessor file
19
  "0-ma/mobilenet-v2-geometric-shapes": "0-ma/mobilenet-v2-geometric-shapes",
20
  "0-ma/focalnet-geometric-shapes-tiny": "0-ma/focalnet-geometric-shapes-tiny" ,
21
  "0-ma/efficientnet-b2-geometric-shapes":"0-ma/efficientnet-b2-geometric-shapes",
 
15
 
16
  # Available models for the dropdown
17
  models = {
18
+ "0-ma/swin-geometric-shapes-tiny": "0-ma/swin-geometric-shapes-tiny",
19
  "0-ma/mobilenet-v2-geometric-shapes": "0-ma/mobilenet-v2-geometric-shapes",
20
  "0-ma/focalnet-geometric-shapes-tiny": "0-ma/focalnet-geometric-shapes-tiny" ,
21
  "0-ma/efficientnet-b2-geometric-shapes":"0-ma/efficientnet-b2-geometric-shapes",