Spaces:
Sleeping
Sleeping
Kaniz-Fatema-Nabila
commited on
Commit
•
1a1b190
1
Parent(s):
2155591
Update app.py
Browse files
app.py
CHANGED
@@ -17,12 +17,13 @@ art_labels= [
|
|
17 |
model = load_learner('models/artStyle_recognizer-v5.pk1')
|
18 |
|
19 |
def recognize_image(image):
|
20 |
-
|
21 |
return dict(zip(art_labels, map(float, probs)))
|
22 |
|
23 |
image = gr.Image()
|
24 |
label = gr.Label()
|
25 |
example = [
|
|
|
26 |
'Constructivism art.jpg',
|
27 |
'Fauvism art.jpg',
|
28 |
'Gothic art.jpeg',
|
|
|
17 |
model = load_learner('models/artStyle_recognizer-v5.pk1')
|
18 |
|
19 |
def recognize_image(image):
|
20 |
+
_, _, probs = model.predict(image)
|
21 |
return dict(zip(art_labels, map(float, probs)))
|
22 |
|
23 |
image = gr.Image()
|
24 |
label = gr.Label()
|
25 |
example = [
|
26 |
+
'Art Nouvaeu.jpg',
|
27 |
'Constructivism art.jpg',
|
28 |
'Fauvism art.jpg',
|
29 |
'Gothic art.jpeg',
|