Update app.py

app.py

@@ -35,13 +35,11 @@ def load_t5():
     model = AutoModelForSeq2SeqLM.from_pretrained("t5-base")
 
     tokenizer = AutoTokenizer.from_pretrained("t5-base")
-    st.success("Loaded T5 Model")
     return model, tokenizer
 
 
 @st.cache
 def custom_model():
-    st.success("Loaded custom model")
     return pipeline("summarization", model="my_awesome_sum/")
 
 
@@ -53,7 +51,6 @@ def convert_df(df):
 
 @st.cache
 def load_one_line_summarizer(model):
-    st.success("Loaded one line summarizer")
     return model.load_model("t5", "snrspeaks/t5-one-line-summary")
 
 
@@ -61,7 +58,6 @@ def load_one_line_summarizer(model):
 def classify_category():
     tokenizer = BertTokenizer.from_pretrained("bert-base-uncased")
     new_model = load_model("model")
-    st.success("Loaded custom classification model")
     return tokenizer, new_model