Edit default value for max tokens and providers formatting

app.py

@@ -40,7 +40,6 @@ Parameters_Col, Chat_Col = st.columns([1,3])
 
 with Parameters_Col:
 
-#    st.header("LLM Router")
     st.image(
          "./assets/unify_logo.png",
          use_column_width="auto",
@@ -63,10 +62,10 @@ with Parameters_Col:
     with st.expander("Advanced Inputs"):
         max_tokens = st.slider(
             label = "Maximum Number Of Tokens",
-            min_value=10,
-            max_value=500,
-            value=100,
-            step=20,
+            min_value=100,
+            max_value=2000,
+            value=500,
+            step=100,
             help = "The maximum number of tokens that can be generated."
         )
         temperature = st.slider(
@@ -137,6 +136,8 @@ with Chat_Col:
             
             # Displaying model, provider, and metrics
             provider = " ".join(chunks[0].model.split("@")[-1].split("-")).title()
+            if " Ai" in provider:
+                provider = provider.replace("Ai", "AI")
             st.markdown(f"Model: **{model}**. Provider: **{provider}**")
             st.markdown(
                 f"**{tokens_per_second}** Tokens Per Second - \
@@ -152,4 +153,4 @@ with Chat_Col:
     # Cancel / Stop button
     if st.button("Clear Chat", key="clear"):
         msgs.empty()
-        st.session_state.messages = []
+        st.session_state.messages = []