Update app.py

app.py

@@ -75,16 +75,14 @@ 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 = []
-        if not empty:
+        else:
+            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:
@@ -92,5 +90,4 @@ with st.form("Taxonomizer"):
                 else:
                     if i == 3:
                         st.subheader('Other possible tags:')
-                    st.caption('* ' + tag_data['tag'] + ' (' + tag_data['description'] + ')')
-    
+                    st.caption('* ' + tag_data['tag'] + ' (' + tag_data['description'] + ')')