Update app.py

app.py

@@ -5,7 +5,7 @@ import numpy as np
 
 model = load_model('Pikachu_and_Raichu.h5')
 
-class_names = ['Pikachu', 'Raichu']
+labels = ['Pikachu', 'Raichu']
 
 def predict(img):
     img=img.reshape(160,160,3)
@@ -13,10 +13,10 @@ def predict(img):
     images_list.append(np.array(img))
     x = np.asarray(images_list)"""
     prediction = model.predict(img)[0]
-    return {class_names[i]: float(prediction[i]) for i in range(len(class_names))}
+    return {labels[i]: float(prediction[i]) for i in range(len(labels))}
 
 image = gr.inputs.Image(shape=(160, 160))
 label = gr.outputs.Label(num_top_classes=2)
 
 gr.Interface(fn=predict, inputs=image,  title="Garbage Classifier",
-    description="This is a Garbage Classification Model Trained using Dataset 11 by Sud.Deployed to Hugging Faces using Gradio.",outputs=label,interpretation='default').launch(debug=True,enable_queue=True)
+    description="Tradio.",outputs=label,interpretation='default').launch(debug=True,enable_queue=True)