Spaces:

chansung
/

flowerclf-space

Runtime error

chansung commited on May 27, 2022

Commit

fa179a8

•

1 Parent(s): 6dbc7c7

Upload app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -21,7 +21,7 @@ def classify_image(inp):
   inp = resnet50.preprocess_input(inp)
   prediction = model.predict(inp).flatten()
-  confidences = {labels[i]: float(prediction[i]) for i in range(1000)}
   return confidences
 iface = gr.Interface(fn=classify_image,

   inp = resnet50.preprocess_input(inp)
   prediction = model.predict(inp).flatten()
+  confidences = {labels[i]: float(prediction[i]) for i in range(10)}
   return confidences
 iface = gr.Interface(fn=classify_image,