czkaiweb commited on
Commit
475c4b5
·
1 Parent(s): e912fcf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -49,12 +49,12 @@ def inference(input_image):
49
  categories = [s.strip() for s in f.readlines()]
50
 
51
  categories = {
52
- 0:vanGogh,
53
- 1:Monet,
54
- 2:Leonardo da Vinci,
55
- 3:Rembrandt,
56
- 4:Pablo Picasso,
57
- 5:Salvador Dali
58
  }
59
 
60
  # Show top categories per image
 
49
  categories = [s.strip() for s in f.readlines()]
50
 
51
  categories = {
52
+ 0:vanGogh,
53
+ 1:Monet,
54
+ 2:Leonardo da Vinci,
55
+ 3:Rembrandt,
56
+ 4:Pablo Picasso,
57
+ 5:Salvador Dali
58
  }
59
 
60
  # Show top categories per image