Update app.py

app.py

@@ -28,8 +28,8 @@ def load_model():
     nltk.download('omw-1.4')
     ## summary_mod_name = os.environ["summary_mod_name"]
     ## question_mod_name = os.environ["question_mod_name"]
-    summary_mod_name = "t5-large"
-    question_mod_name = "t5-large"
+    summary_mod_name = "csebuetnlp/mT5_multilingual_XLSum"
+    question_mod_name = "MaRiOrOsSi/t5-base-finetuned-question-answering"
     summary_model = T5ForConditionalGeneration.from_pretrained(summary_mod_name)
     summary_tokenizer = T5Tokenizer.from_pretrained(summary_mod_name)
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
@@ -43,17 +43,17 @@ from nltk.corpus import wordnet as wn
 from nltk.tokenize import sent_tokenize
 from nltk.corpus import stopwords
 
-def csv_downloader(df):
-    res = df.to_csv(index=False,sep="\t").encode('utf-8')
-    st.download_button(
-    label="Download logs data as CSV separated by tab",
-    data=res,
-    file_name='df_quiz_log_file_v1.csv',
-    mime='text/csv')
+# def csv_downloader(df):
+#    res = df.to_csv(index=False,sep="\t").encode('utf-8')
+#    st.download_button(
+#    label="Download logs data as CSV separated by tab",
+#    data=res,
+#    file_name='df_quiz_log_file_v1.csv',
+#    mime='text/csv')
 
 def load_file():
     """Load text from file"""
-    uploaded_file = st.file_uploader("Upload Files",type=['txt'])
+    uploaded_file = st.file_uploader("Paste text",type=['txt'])
     if uploaded_file is not None:
         if uploaded_file.type == "text/plain":
             raw_text = str(uploaded_file.read(),"utf-8")
@@ -64,11 +64,11 @@ def load_file():
 summary_model, summary_tokenizer, question_tokenizer, question_model =load_model()
 
 # App title and description
-st.title("Exam Assistant")
-st.write("Upload text, Get ready for answering autogenerated questions")
+st.title("P's Prototye")
+st.write("Get multiple choice questions from random facts")
 
 # Load file
-st.text("Disclaimer: This app stores user's input for model improvement purposes !!")
+st.text("Disclaimer: This is early version. sorry if there's still bugs")
 
 # Load file