EvanMarie commited on
Commit
d54de78
·
1 Parent(s): 66f301a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import gradio as gr
5
 
6
 
7
  learner_attractiveness = load_learner('hotornot.pk1')
8
- categories = ['hot', 'average', 'ugly']
9
 
10
  def classify_attractiveness(img):
11
  prediction, index, probability = learner_attractiveness.predict(img)
 
5
 
6
 
7
  learner_attractiveness = load_learner('hotornot.pk1')
8
+ categories = ['average','hot', 'ugly']
9
 
10
  def classify_attractiveness(img):
11
  prediction, index, probability = learner_attractiveness.predict(img)