Update app.py

app.py

@@ -14,10 +14,10 @@ pre_trained_model = from_pretrained_keras(model_api_link)
 labels = ['compost', 'e-waste', 'recycle', 'trash']
 
 
-def classify_image(input):
-    input = input.reshape((1, 224, 224, 3))
-    input = tf.keras.applications.mobilenet_v2.preprocess_input(input)
-    prediction = pre_trained_model.predict(input) 
+def classify_image(inp):
+    inp = inp.reshape((-1, 224, 224, 3))
+    #inp = tf.keras.applications.mobilenet_v2.preprocess_input(inp)
+    prediction = model.predict(inp).flatten()
     confidences = {labels[i]: float(prediction[i]) for i in range(4)}
     return confidences
 
@@ -25,7 +25,7 @@ def classify_image(input):
 iface = gr.Interface(fn=classify_image, 
              inputs=gr.Image(),
              outputs=gr.Label(num_top_classes=4),
-             examples=["banana.jpg", "battery.jpg", "can.jpg", "paper.jpg"]
+             examples=["banana.jpg", "battery.jpg", "can.jpg"]
              )
              
 iface.launch(share=True)