Update app.py

app.py

@@ -10,7 +10,7 @@ classnames = ['cardboard', 'metal','paper','plastic','trash','green-glass','whit
 def predict(img):
     img=img.reshape(-1,298, 384,3)
     prediction = model.predict(img[0])
-    return {classnames[i]: float(prediction[i]) for i in range(12)}
+    return {classnames[i]: float(prediction[i]) for i in range(len(classnames)}
 
 image = gr.inputs.Image(shape=(298, 384))
 label = gr.outputs.Label(num_top_classes=3)