Update app.py

app.py

@@ -5,17 +5,19 @@ import numpy as np
 
 model = load_model('Pikachu_and_Raichu.h5')
 
-#class_names[labels] = ['Pikachu', 'Raichu']
+labels = ['Pikachu', 'Raichu']
 
 def predict(img):
-    img=img.reshape(160,160,3)
+    img=img.reshape(-1,224,224,3)
     """images_list = []
     images_list.append(np.array(img))
     x = np.asarray(images_list)"""
-    prediction = model.predict(img)[0]
-    return {class_names[labels[i]]: float(prediction[i]) for i in range(len(class_names[labels]))}
+    #prediction = model.predict(img)[0]
+    prediction = model.predict(img).flatten()
+    confidences = {labels[i]: float(prediction[i]) for i in range(2)}
+    return confidences
 
-image = gr.inputs.Image(shape=(160, 160))
+image = gr.inputs.Image(shape=(224, 224))
 label = gr.outputs.Label(num_top_classes=2)
 
 gr.Interface(fn=predict, inputs=image,  title="Garbage Classifier",