0-ma commited on
Commit
5dcea5c
·
verified ·
1 Parent(s): e7f2b5e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -11,8 +11,9 @@ labels = [
11
  "Pentagon",
12
  "Hexagon"
13
  ]
14
- #images = [Image.open(requests.get("https://raw.githubusercontent.com/0-ma/geometric-shape-detector/main/input/exemple_circle.jpg", stream=True).raw),
15
- # Image.open(requests.get("https://raw.githubusercontent.com/0-ma/geometric-shape-detector/main/input/exemple_pentagone.jpg", stream=True).raw)]
 
16
  feature_extractor = AutoImageProcessor.from_pretrained('0-ma/vit-geometric-shapes-tiny')
17
  model = AutoModelForImageClassification.from_pretrained('0-ma/vit-geometric-shapes-tiny')
18
 
 
11
  "Pentagon",
12
  "Hexagon"
13
  ]
14
+
15
+
16
+
17
  feature_extractor = AutoImageProcessor.from_pretrained('0-ma/vit-geometric-shapes-tiny')
18
  model = AutoModelForImageClassification.from_pretrained('0-ma/vit-geometric-shapes-tiny')
19