Upload settings.py

settings.py

@@ -0,0 +1,95 @@
+import streamlit as st
+
+settings = {}
+
+def app():
+    st.markdown("""
+        <style>
+            div[data-testid="stForm"] {
+                border: 0;
+            }
+            .footer-custom {
+                position: fixed;
+                bottom: 0;
+                width: 100%;
+                color: var(--text-color);
+                max-width: 698px;
+                font-size: 14px;
+                height: 50px;
+                padding: 10px 0;
+                z-index: 50;
+            }
+            footer {
+                display: none !important;
+            }
+            .footer-custom a {
+                color: var(--text-color);
+            }
+            button[kind="formSubmit"]{
+                margin-top: 40px;
+                border-radius: 20px;
+                padding: 5px 20px;
+                font-size: 18px;
+                background-color: var(--primary-color);
+            }
+            #lfqa-model-parameters {
+                margin-bottom: 50px;
+                font-size: 36px;
+            }
+            #tts-model-parameters {
+                font-size: 36px;
+                margin-top: 50px;
+            }
+            .stAlert {
+                width: 250px;
+                margin-top: 32px;
+            }
+        </style>
+    """, unsafe_allow_html=True)
+
+    with st.form("settings"):
+        footer = """
+            <div class="footer-custom">
+                Streamlit app - <a href="https://www.linkedin.com/in/danijel-petkovic-573309144/" target="_blank">Danijel Petkovic</a>  |   
+                LFQA/DPR models - <a href="https://www.linkedin.com/in/blagojevicvladimir/" target="_blank">Vladimir Blagojevic</a>   |
+                Guidance & Feedback - <a href="https://yjernite.github.io/" target="_blank">Yacine Jernite</a>
+            </div>
+        """
+        st.markdown(footer, unsafe_allow_html=True)
+
+        st.title("LFQA model parameters")
+
+        settings["min_length"] = st.slider("Min length", 20, 80, st.session_state["min_length"],
+                                           help="Min response length (words)")
+        st.markdown("""<hr></hr>""", unsafe_allow_html=True)
+        settings["max_length"] = st.slider("Max length", 128, 320, st.session_state["max_length"],
+                                           help="Max response length (words)")
+        st.markdown("""<hr></hr>""", unsafe_allow_html=True)
+        col1, col2 = st.columns(2)
+        with col1:
+            settings["do_sample"] = st.checkbox("Use sampling", st.session_state["do_sample"],
+                                                help="Whether or not to use sampling ; use greedy decoding otherwise.")
+        with col2:
+            settings["early_stopping"] = st.checkbox("Early stopping", st.session_state["early_stopping"],
+                                                    help="Whether to stop the beam search when at least num_beams sentences are finished per batch or not.")
+        st.markdown("""<hr></hr>""", unsafe_allow_html=True)
+        settings["num_beams"] = st.slider("Num beams", 1, 16, st.session_state["num_beams"],
+                                          help="Number of beams for beam search. 1 means no beam search.")
+        st.markdown("""<hr></hr>""", unsafe_allow_html=True)
+        settings["temperature"] = st.slider("Temperature", 0.0, 1.0, st.session_state["temperature"], step=0.1,
+                                            help="The value used to module the next token probabilities")
+
+        st.title("TTS model parameters")
+        settings["tts"] = st.selectbox(label="Engine", options=("Google", "HuggingFace"),
+                                       index=["Google", "HuggingFace"].index(st.session_state["tts"]),
+                                       help="Answer text-to-speech engine")
+
+        # Every form must have a submit button.
+        col3, col4, col5, col6 = st.columns(4)
+        with col3:
+            submitted = st.form_submit_button("Save")
+        with col4:
+            if submitted:
+                for k, v in settings.items():
+                    st.session_state[k] = v
+                st.success('App settings saved successfully.')