Update app.py

app.py

@@ -1,12 +1,10 @@
-__all__ = ['learn', 'classify_image', 'categories', 
-           'image', 'label', 'examples', 'intf']
 
 from fastai.vision.all import *
 import gradio as gr
 
-learn = load_learner('model.pkl')
+learn=load_learner('sportball.pkl')
 
-categories = ('Dinosaur', 'Not Dinosaur')
+categories = ['soccer ball','basketball ball','golf ball','tennis ball','volleyball ball','Cricket ball']
 
 def classify_image(img):
     pred, idx, probs = learn.predict(img)
@@ -14,7 +12,14 @@ def classify_image(img):
 
 image = gr.inputs.Image(shape=(192, 192))
 label = gr.outputs.Label()
-examples = ['dinosaur.png', 'not_dinosaur.png']
+examples = [
+    'soccer_ball.jpg',
+    'basketball_ball.jpg',
+    'volleyball.jpg',
+    'tennis_ball.jpg',
+    'golf_ball.jpg',
+    'cricket_ball.jpg'
+]
 
-intf = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)
+intf = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples, title = "Sports Ball Recognition App", description = "This app can recognize different types of sport balls.")
 intf.launch(inline=False)