Update app.py

app.py

@@ -16,7 +16,7 @@ abstract = st.text_area("Enter the abstract of the article", height=350)
 top_k = st.slider('How many topics from top to show?', 1, 10, 3)
 text = title + " " + abstract
 
-@st.cache(suppress_st_warning=True)
+@st.cache(suppress_st_warning=False)
 def get_model_tokenizer_linear():
     tokenizer = DistilBertTokenizer.from_pretrained("distilbert-base-cased")
     model = DistilBertModel.from_pretrained("distilbert-base-cased")
@@ -26,5 +26,8 @@ def get_model_tokenizer_linear():
     my_linear.load_state_dict(torch.load(MY_LINEAR_NAME, map_location=torch.device('cpu')))
     return {"model": model, "tokenizer": tokenizer, "my_linear": my_linear}
 
-for ms in get_answer_with_desc(text, top_k=top_k, **get_model_tokenizer_linear()):
-    st.markdown("#### " + ms)
+if len(text) == 1:
+      st.markdown("Input is empty, write something!")  
+else:
+    for ms in get_answer_with_desc(text, top_k=top_k, **get_model_tokenizer_linear()):
+        st.markdown("#### " + ms)