Update app.py

app.py

@@ -75,18 +75,22 @@ with st.form("Taxonomizer"):
     submitted = st.form_submit_button("Taxonomize")
     st.caption('We **do not** recommend using ArXivTaxonomizer© to choose tags for you new paper.')
     if submitted:
+    empty = False
         if title == '':
             st.markdown("You are most definitely abusing our service. No ticket today, but you better be careful.")
+            empty = True
         prompt = 'Title: ' + title + ' Abstract: ' + abstract
         tokens = tokenizer(prompt, truncation=True, padding='max_length', return_tensors='pt')['input_ids']
         preds = softmax(model(tokens.reshape(1, -1)).logits / temperature, dim=1)
         tags = top_xx(preds, xx)
         other_tags = []
-        st.header('Inferred tags:')
-        for i, tag_data in enumerate(tags):
-            if i < 3:
-                st.markdown('* ' + tag_data['tag'] + ' (' + tag_data['description'] + ')')
-            else:
-                if i == 3:
-                    st.subheader('Other possible tags:')
-                st.caption('* ' + tag_data['tag'] + ' (' + tag_data['description'] + ')')
+        if not empty:
+            st.header('Inferred tags:')
+            for i, tag_data in enumerate(tags):
+                if i < 3:
+                    st.markdown('* ' + tag_data['tag'] + ' (' + tag_data['description'] + ')')
+                else:
+                    if i == 3:
+                        st.subheader('Other possible tags:')
+                    st.caption('* ' + tag_data['tag'] + ' (' + tag_data['description'] + ')')
+