Update app.py

app.py

@@ -78,7 +78,7 @@ def main():
         probs = outputs.logits.softmax(dim=1)
         percentage = round(probs[0, label_num].item() * 100, 2)
 
-        st.write("The prediction class is:")
+        st.write("The Predicted Classification is:")
 
         if label_num==0:
             st.write("Bread (" + f"{percentage}%)")
@@ -106,11 +106,11 @@ def main():
             st.write("Vegetable (" + f"{percentage}%)")
 
     
-        select_health = st.selectbox("Select one:", ["Choose healthy or non-healthy", "Healthy", "Non-healthy"])
+        select_health = st.selectbox("Select one:", ["Choose Healthy or Non-Healthy (Not Functional Yet)", "Healthy", "Non-Healthy"])
         if select_health == "Healthy":
             st.write("You selected healthy for", "Bread" if label_num==0 else "Dairy" if label_num==1 else "Dessert" if label_num==2 else "Egg" if label_num==3 else "Fried Food" if label_num==4 else "Fruit" if label_num==5 else "Meat" if label_num==6 else "Noodles" if label_num==7 else "Rice" if label_num==8 else "Seafood" if label_num==9 else "Soup" if label_num==10 else "Vegetable")
             # Add code to fetch healthy recipe here
-        elif select_health == "Non-healthy":
+        elif select_health == "Non-Healthy":
             st.write("You selected non-healthy for", "Bread" if label_num==0 else "Dairy" if label_num==1 else "Dessert" if label_num==2 else "Egg" if label_num==3 else "Fried Food" if label_num==4 else "Fruit" if label_num==5 else "Meat" if label_num==6 else "Noodles" if label_num==7 else "Rice" if label_num==8 else "Seafood" if label_num==9 else "Soup" if label_num==10 else "Vegetable")
             # Add code to fetch unhealthy recipe here