Modify the table truncation logic

app.py

@@ -10,7 +10,7 @@ st.set_page_config(
     menu_items=None,
 )
 
-with st.spinner("Setting stuff up..."):
+with st.spinner("Setting stuff up related to the inference engine..."):
     classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
 
 st.title("Zero-shot classification from tabular data")
@@ -21,12 +21,15 @@ st.text(
 data = st.file_uploader("Upload Excel file (it should contain a `text` column):")
 labels = st.text_input("Enter comma-separated labels:")
 
+# classify first N snippets only for faster inference
+N = 100
+
 if st.button("Calculate labels"):
 
     try:
         labels_list = labels.split(",")
         table = pd.read_excel(data)
-        table = table.loc[table["text"].apply(len) > 10].reset_index(drop=True).head(50)
+        table = table.loc[table["text"].apply(len) > 10].reset_index(drop=True).head(N)
 
         prog_bar = st.progress(0)
         preds = []
@@ -41,5 +44,5 @@ if st.button("Calculate labels"):
 
     except:
         st.error(
-            "File load didn't work. Make sure you upload a file containing a `text` column and a set of comma-separated labels is provided"
+            "Something went wrong. Make sure you upload an Excel file containing a `text` column and a set of comma-separated labels is provided"
         )