Update app.py

app.py

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