ADD: Second model as well as info card about training subsets

app.py

@@ -30,12 +30,14 @@ def main():
     st.text("")
     models_to_choose = [
         "AryanLala/autonlp-Scientific_Title_Generator-34558227",
+        "shamikbose89/mt5-small-finetuned-arxiv-cs-finetuned-arxiv-cs-full"
     ]
 
     BASE_MODEL = st.selectbox("Choose a model to generate the title", models_to_choose)
 
     def preprocess(text):
-        if BASE_MODEL == "AryanLala/autonlp-Scientific_Title_Generator-34558227":
+        if ((BASE_MODEL == "AryanLala/autonlp-Scientific_Title_Generator-34558227") |
+          (BASE_MODEL == "shamikbose89/mt5-small-finetuned-arxiv-cs-finetuned-arxiv-cs-full")):
             return [text]
         else:
             st.error("Please select a model first")
@@ -84,7 +86,24 @@ def main():
     if st.button("Submit"):
         submit(message)
 
-
+    with st.expander("Additional Model Info"):
+        st.write("""
+        The models used were fine-tuned on subset of data from the [Arxiv Dataset](https://huggingface.co/datasets/arxiv_dataset)
+        The task of the models is to suggest an appropraite title from the abstract of a scientific paper.
+        
+        The model [AryanLala/autonlp-Scientific_Title_Generator-34558227]() was trained on data
+        from the Cs.AI (Artificial Intelligence) category of papers.
+
+        The model [shamikbose89/mt5-small-finetuned-arxiv-cs-finetuned-arxiv-cs-full](https://huggingface.co/shamikbose89/mt5-small-finetuned-arxiv-cs-finetuned-arxiv-cs-full) 
+        was trained on the categories: cs.AI, cs.LG, cs.NI, cs.GR cs.CL, cs.CV (Artificial Intelligence, Machine Learning, Networking and Internet Architecture, Graphics, Computation and Language, Computer Vision and Pattern Recognition)
+        """)
+
+    st.text('')
+    st.markdown(
+    '''<span style="color:blue; font-size:10px">App created by [@AlekseyDorkin](https://huggingface.co/AlekseyDorkin)
+     and [@akshay7](https://huggingface.co/akshay7)</span>''', 
+     unsafe_allow_html=True,
+    )
 
 if __name__ == "__main__":
     main()