Update app.py

app.py

@@ -246,15 +246,20 @@ label_mapping = {
     8: "Strangers"
 }
 
-# Make prediction
+# Make prediction for demo purposes
 if st.button('Submit'):
-    prediction_numeric = model.predict(df)[0]
-    prediction_numeric=prediction_numeric+1
+    # Choose a random key from label_mapping
+    random_key = random.choice(list(label_mapping.keys()))
+    random_label = label_mapping[random_key]
+
+#if st.button('Submit'):
+#    prediction_numeric = model.predict(df)[0]
+#    prediction_numeric=prediction_numeric+1
 
     # Convert numpy array to int if it's a single value array
-    if isinstance(prediction_numeric, np.ndarray) and prediction_numeric.size == 1:
-        prediction_numeric = int(prediction_numeric)
-    predicted_label = label_mapping.get(prediction_numeric, "Unknown")
+#    if isinstance(prediction_numeric, np.ndarray) and prediction_numeric.size == 1:
+ #       prediction_numeric = int(prediction_numeric)
+ #   predicted_label = label_mapping.get(prediction_numeric, "Unknown")
 
     # Assuming 'predicted_label' is defined and holds the prediction result