Kaniz-Fatema-Nabila commited on
Commit
1a1b190
1 Parent(s): 2155591

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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
- pred, idx, 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
  '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',