Update app.py

app.py

@@ -66,11 +66,11 @@ if(1):
             if predicted_class:
                 st.write("### **Model 1 Prediction:**")
                 st.write(f"**Predicted Class:** {predicted_class}")
-                st.json({class_names[i]: round(float(prob), 4) for i, prob in enumerate(prediction_probs[0])})
+                st.json({class_names[i]: round(float(prob), 4) for i, prob in enumerate(prediction_probabilities[0])})
 
                 st.write("### **Model 2 Prediction:**")
                 st.write(f"**Predicted Class:** {predicted_class_2}")
-                st.json({class_names[i]: round(float(prob), 4) for i, prob in enumerate(prediction_probs_2[0])})
+                st.json({class_names[i]: round(float(prob), 4) for i, prob in enumerate(prediction_probabilities_2[0])})
 
     
 else: