EvanMarie commited on
Commit
a3b6e30
1 Parent(s): 370267b

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 = ['average','hot', 'not hot']
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', 'or not']
9
 
10
  def classify_attractiveness(img):
11
  prediction, index, probability = learner_attractiveness.predict(img)