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

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