Spaces:

EvanMarie
/

madness_combat

Runtime error

EvanMarie commited on Nov 16, 2022

Commit

1e6dcf6

1 Parent(s): ceacbbe

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,7 +2,7 @@ from fastai.vision.all import *
 import gradio as gr
 learner_madness = load_learner('madness.pkl')
-categories = ['Deimos', 'The Auditor', 'Hank', "Jeb", 'Sanford', 'Tricky']
 def classify_madness(img):
   prediction, index, probability = learner_madness.predict(img)

 import gradio as gr
 learner_madness = load_learner('madness.pkl')
+categories = ['The Auditor', 'Deimos', 'Hank', "Jeb", 'Sanford', 'Tricky']
 def classify_madness(img):
   prediction, index, probability = learner_madness.predict(img)