translate, summarize, similarity

app.py

@@ -6,7 +6,6 @@ import tensorflow_text
 import tensorflow as tf
 def main():
     st.title("Interactive demo: T5 Multitasking Demo")
-    st.sidebar.image("https://i.gzn.jp/img/2020/02/25/google-ai-t5/01.png")
     saved_model_path = load_model_cache()
     # Model is loaded in st.session_state to remain stateless across reloading
     if 'model' not in st.session_state:
@@ -20,7 +19,7 @@ def load_model_cache():
     if not os.path.exists(CACHE_DIR):
         os.mkdir(CACHE_DIR)
     # download the files from huggingface repo and load the model with tensorflow
-    snapshot_download(repo_id="stevekola/T5", cache_dir=CACHE_DIR)
+    snapshot_download(repo_id="widged/translate-to-french", cache_dir=CACHE_DIR)
     saved_model_path = os.path.join(CACHE_DIR, os.listdir(CACHE_DIR)[0])
     return saved_model_path
 def dashboard(model):
@@ -31,9 +30,6 @@ def dashboard(model):
     task_type = st.sidebar.radio("Task Type",
                                  [
                                      "Translate English to French",
-                                     "Translate English to German",
-                                     "Translate English to Romanian",
-                                     "Grammatical Correctness of Sentence",
                                      "Text Summarization",
                                      "Document Similarity Score"
                                  ])
@@ -107,9 +103,6 @@ def predict(task_type, sentence, model):
     """
     task_dict = {
         "Translate English to French": "Translate English to French",
-        "Translate English to German": "Translate English to German",
-        "Translate English to Romanian": "Translate English to Romanian",
-        "Grammatical Correctness of Sentence": "cola sentence",
         "Text Summarization": "summarize",
         "Document Similarity Score": "stsb",
     }