Spaces:

cvergnes
/

bmw_or_mercedes

Build error

cvergnes commited on Jul 30, 2022

Commit

60ced6c

•

1 Parent(s): 5de835d

wip

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ categories = ['bmw car', 'mercedes car']
 learn = load_learner("export.pkl")
 def classify_image(img):
-  pred, _, prob = learn.predict(img)
   return dict(zip(categories, map(float, probs)))
 intf = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)

 learn = load_learner("export.pkl")
 def classify_image(img):
+  pred, _, probs = learn.predict(img)
   return dict(zip(categories, map(float, probs)))
 intf = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)