Spaces:

NikiTricky
/

resnet-food101

Sleeping

NikiTricky commited on Jun 1, 2023

Commit

bde552e

1 Parent(s): d49f24f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ def classify_image(inp):
   print("Classifying image...")
   inp = inp.reshape((-1, 224, 224, 3))
   inp == inp/255
-  prediction = model.predict(inp)
   confidences = {model.config['id2label'][str(i)]: float(prediction[i]) for i in range(101)}
   return confidences

   print("Classifying image...")
   inp = inp.reshape((-1, 224, 224, 3))
   inp == inp/255
+  prediction = model.predict(inp)[0]
   confidences = {model.config['id2label'][str(i)]: float(prediction[i]) for i in range(101)}
   return confidences