Update app.py

app.py

@@ -43,8 +43,8 @@ def Analysing_image(st, model, image_file):
 
         # Assuming a multi-class classification where the class with the highest probability is selected:
         predicted_class = np.argmax(predictions, axis=1)
-        st.write("Prediction :", keys(str(predicted_class[0])))
-        return keys(str(predicted_class[0]))
+        st.write("Prediction :", keys[str(predicted_class[0])])
+        return keys[str(predicted_class[0])]
     except:
         return None