Update app.py

app.py

@@ -126,22 +126,23 @@ model_2 = load_GlotLID_v2(constants.MODEL_NAME, "model_v2.bin")
 def plot(label, prob):
 
     ORANGE_COLOR = "#FF8000"
+    BLACK_COLOR = "#000000"
     fig, ax = plt.subplots(figsize=(8, 1))
     fig.patch.set_facecolor("none")
     ax.set_facecolor("none")
 
     ax.spines["left"].set_color(ORANGE_COLOR)
     ax.spines["bottom"].set_color(ORANGE_COLOR)
-    ax.tick_params(axis="x", colors=ORANGE_COLOR)
+    ax.tick_params(axis="x", colors=BLACK_COLOR)
 
     ax.spines[["right", "top"]].set_visible(False)
 
     ax.barh(y=[0], width=[prob], color=ORANGE_COLOR)
     ax.set_xlim(0, 1)
     ax.set_ylim(-1, 1)
-    ax.set_title(f"Label: {label}, Language: {get_name(label)}", color=ORANGE_COLOR)
+    ax.set_title(f"Label: {label}, Language: {get_name(label)}", color=BLACK_COLOR)
     ax.get_yaxis().set_visible(False)
-    ax.set_xlabel("Confidence", color=ORANGE_COLOR)
+    ax.set_xlabel("Confidence", color=BLACK_COLOR)
     st.pyplot(fig)
 
 def compute(sentences, version = 'v2'):