Update app.py

app.py

@@ -291,8 +291,6 @@ def display_dishes_in_grid(dishes, cols=3):
                 st.sidebar.write(dish.replace("_", " ").capitalize())
 
 def display_prediction_graph(class_names, confidences):
-    theme_mode = st.get_option("theme.base")
-    print("ThEME MODE IQEWOKWQPIJWQPIJEWQOPIJWQPIJEWQPIJEWQPJIEWQPWQEPEWQPOJPJQWEOPJW",theme_mode)
     #reversing them so graph displays highest predictions at the top
     confidences.reverse()
     class_names.reverse()
@@ -315,7 +313,7 @@ def display_prediction_graph(class_names, confidences):
         if width < min_width:
             width = min_width
         ax.text(width - 0.02, bar.get_y() + bar.get_height()/2, f'{original_width:.1f}%', 
-                va='center', ha='right', color='white' if theme_mode == "dark" else "black", fontweight='bold', fontsize=16)
+                va='center', ha='right', color='white', fontweight='bold', fontsize=16)
 
     ax.set_xticklabels([]) #remove x label
 
@@ -323,13 +321,13 @@ def display_prediction_graph(class_names, confidences):
     max_label_width = 10
     labels = ax.get_yticklabels()
     wrapped_labels = [textwrap.fill(label.get_text(), width=max_label_width) for label in labels] # Wrap the labels if they exceed the max width
-    ax.set_yticklabels(wrapped_labels, fontsize=16, color='white' if theme_mode == "dark" else "black")
+    ax.set_yticklabels(wrapped_labels, fontsize=16, color='white')
     
     #no borders
     for spine in ax.spines.values():
         spine.set_visible(False)
 
-    ax.set_title(class_names[-1], color='white' if theme_mode == "dark" else "black", fontsize=24, fontweight='bold', ha='left', x=0.5)
+    ax.set_title(class_names[-1], color='white', fontsize=24, fontweight='bold', ha='left', x=0.5)
     
     st.pyplot(fig) # Display the plot