Update app.py

app.py

@@ -3,17 +3,25 @@ import random
 import requests
 from mtranslate import translate
 import streamlit as st
-MODEL_URL = "https://api-inference.huggingface.co/models/flax-community/papuGaPT2"
+LOGO = "https://raw.githubusercontent.com/nlp-en-es/assets/main/logo.png"
+MODELS = {
+    "Model trained on OSCAR": {
+        "url": "https://api-inference.huggingface.co/models/flax-community/gpt-2-spanish"
+    },
+    "Model trained on the Large Spanish Corpus": {
+        "url": "https://api-inference.huggingface.co/models/mrm8488/spanish-gpt2"
+    },
+}
 PROMPT_LIST = {
-    "Najsmaczniejszy owoc to...": ["Najsmaczniejszy owoc to "],
-    "Cześć, mam na imię...": ["Cześć, mam na imię "],
-    "Największym polskim poetą był...": ["Największym polskim poetą był "],
+    "Érase una vez...": ["Érase una vez "],
+    "¡Hola!": ["¡Hola! Me llamo "],
+    "¿Ser o no ser?": ["En mi opinión, 'ser' es "],
 }
-def query(payload, model_url):
+def query(payload, model_name):
     data = json.dumps(payload)
-    print("model url:", model_url)
+    print("model url:", MODELS[model_name]["url"])
     response = requests.request(
-        "POST", model_url, headers={}, data=data
+        "POST", MODELS[model_name]["url"], headers={}, data=data
     )
     return json.loads(response.content.decode("utf-8"))
 def process(
@@ -34,9 +42,10 @@ def process(
     }
     return query(payload, model_name)
 # Page
-st.set_page_config(page_title="papuGaPT2 (Polish GPT-2) Demo")
-st.title("papuGaPT2 (Polish GPT-2")
+st.set_page_config(page_title="Spanish GPT-2 Demo", page_icon=LOGO)
+st.title("Spanish GPT-2")
 # Sidebar
+st.sidebar.image(LOGO)
 st.sidebar.subheader("Configurable parameters")
 max_len = st.sidebar.number_input(
     "Maximum length",
@@ -68,12 +77,19 @@ do_sample = st.sidebar.selectbox(
 # Body
 st.markdown(
     """
-    papuGaPT2 (Polish GPT-2) model trained from scratch on OSCAR dataset.
+    Spanish GPT-2 models trained from scratch on two different datasets. One
+    model is trained on the Spanish portion of
+    [OSCAR](https://huggingface.co/datasets/viewer/?dataset=oscar)
+    and the other on the
+    [large_spanish_corpus](https://huggingface.co/datasets/viewer/?dataset=large_spanish_corpus)
+    aka BETO's corpus.
     
-    The models were trained with Jax and Flax using TPUs as part of the [Flax/Jax Community Week](https://discuss.huggingface.co/t/open-to-the-community-community-week-using-jax-flax-for-nlp-cv/7104) organised by HuggingFace.
+    The models are trained with Flax and using TPUs sponsored by Google since this is part of the
+    [Flax/Jax Community Week](https://discuss.huggingface.co/t/open-to-the-community-community-week-using-jax-flax-for-nlp-cv/7104)
+    organised by HuggingFace.
     """
 )
-model_name = MODEL_URL
+model_name = st.selectbox("Model", (list(MODELS.keys())))
 ALL_PROMPTS = list(PROMPT_LIST.keys()) + ["Custom"]
 prompt = st.selectbox("Prompt", ALL_PROMPTS, index=len(ALL_PROMPTS) - 1)
 if prompt == "Custom":
@@ -107,10 +123,24 @@ if st.button("Run"):
                         st.write(f"{error}")
         else:
             result = result[0]["generated_text"]
-            st.write(result.replace("\
-", "  \
-"))
+            st.write(result.replace("\n", "  \n"))
             st.text("English translation")
-            st.write(translate(result, "en", "es").replace("\
-", "  \
-"))
+            st.write(translate(result, "en", "es").replace("\n", "  \n"))
+st.markdown(
+    """
+    ### Team members
+    - Manuel Romero ([mrm8488](https://huggingface.co/mrm8488))
+    - María Grandury ([mariagrandury](https://huggingface.co/mariagrandury))
+    - Pablo González de Prado ([Pablogps](https://huggingface.co/Pablogps))
+    - Daniel Vera ([daveni](https://huggingface.co/daveni))
+    - Sri Lakshmi ([srisweet](https://huggingface.co/srisweet))
+    - José Posada ([jdposa](https://huggingface.co/jdposa))
+    - Santiago Hincapie ([shpotes](https://huggingface.co/shpotes))
+    - Jorge ([jorgealro](https://huggingface.co/jorgealro))
+    
+    ### More information
+    You can find more information about these models in their cards:
+    - [Model trained on OSCAR](https://huggingface.co/models/flax-community/gpt-2-spanish)
+    - [Model trained on the Large Spanish Corpus](https://huggingface.co/mrm8488/spanish-gpt2)
+    """
+)