EvanMarie commited on
Commit
847e104
·
1 Parent(s): 1bf01e5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ import gradio as gr
3
 
4
 
5
  learner = load_learner('anime_classifier.pkl')
6
- categories = anime_characters = ['Dio Brando', 'Hatsune Miku', 'Enrico Pucci', 'Jotaro Kujo', 'Madara Uchiha', 'Princess Mononoke', 'Totoro']
7
 
8
  def classify_anime(img):
9
  prediction, index, probability = learner.predict(img)
 
3
 
4
 
5
  learner = load_learner('anime_classifier.pkl')
6
+ categories = anime_characters = ['Dio Brando', 'Enrico Pucci', 'Hatsune Miku', 'Jotaro Kujo', 'Madara Uchiha', 'Princess Mononoke', 'Totoro']
7
 
8
  def classify_anime(img):
9
  prediction, index, probability = learner.predict(img)