Update app.py

app.py

@@ -7,16 +7,14 @@ labels = ['american_football', 'baseball', 'basketball', 'billiard_ball', 'bowli
                      'football', 'golf_ball', 'hockey_ball', 'hockey_puck', 'rugby_ball', 'shuttlecock',
                      'table_tennis_ball', 'tennis_ball', 'volleyball']
 
-def image_classifier(inp):
+def predict_image(inp):
     img = tf.image.resize(inp, (224, 224))
     img = tf.keras.preprocessing.image.img_to_array(img)
     img = np.expand_dims(img, axis=0)
     
-    predictions = model.predict(img)[0]
+    prediction = model.predict(img)[0]
     
-    return {label[i]: float(predictions[i]) for i in range(25)}
-    
-   
+    return {labels[i]: float(prediction[i]) for i in range(len(labels))}
 
-demo = gr.Interface(fn=image_classifier, inputs='image',outputs=gr.Label(num_top_classes=5),title='Indian Bird Classification')
+demo = gr.Interface(fn=predict_image, inputs='image',outputs=gr.Label(num_top_classes=3),title='Sports Ball Classification')
 demo.launch(debug=True,share=True)