Update app.py

app.py

@@ -1,20 +1,17 @@
-import numpy as np # linear algebra
-import pandas as pd # data processing, CSV file I/O (e.g. pd.read_csv)
-
 import os
 for dirname, _, filenames in os.walk('/kaggle/input'):
     for filename in filenames:
         print(os.path.join(dirname, filename))
 
 from fastai.vision.all import *
+!pip install gradio
 import gradio as gr
 
 share = True
-def is_cat(x): return x[0].isupper()
 
-learn = load_learner('model.pkl')
+learn = load_learner('wastedetective.pkl')
 
-categories = ('Batteries', 'Organic Matter', 'Primary', 'Tertiary')
+categories = ('Batteries', 'Organic Matter', 'Primary', 'Tertiary', 'Metals')
 
 def classify_image(img):
     pred,idx,probs = learn.predict(img)
@@ -23,5 +20,5 @@ def classify_image(img):
 image = "image"
 label = "label"
 
-intf = gr.Interface(fn=classify_image, inputs=image, outputs=label)
-intf.launch(inline=False, share = True)
+intf = gr.Interface(fn=classify_image, inputs = image, outputs = label)
+intf.launch(inline = False, share = True)