Update app.py

app.py

@@ -51,6 +51,8 @@ def model(img):
 
     predicted_class_index = np.argmax(predictions[0])
     print(predicted_class_index)
+
+    predicted_class_label = class_names[predicted_class_index]
     
     return predicted_class_label