Update app.py

app.py

@@ -12,7 +12,9 @@ from langchain.retrievers import BM25Retriever, EnsembleRetriever
 from langchain.schema import Document
 from langchain.vectorstores import Chroma
 from langchain.embeddings import HuggingFaceEmbeddings
-
+from setfit import SetFitModel
+# Download from the 🤗 Hub
+clause_model = SetFitModel.from_pretrained("scholarly360/setfit-contracts-clauses")
 
 def util_upload_file_and_return_list_docs(uploaded_files):
     #util_del_cwd()
@@ -130,10 +132,21 @@ with st.form("my_form"):
     list_save_path = []
     uploaded_files = st.file_uploader("Choose file(s)", accept_multiple_files=True)
     print('uploaded_files ', uploaded_files)
-    submitted = st.form_submit_button("Index and Calculate")
+    submitted = st.form_submit_button("Calculate")
+
+    my_list_structure = []
+    import pandas as pd
     
     if submitted and (uploaded_files is not None):
         list_docs, list_save_path = util_upload_file_and_return_list_docs(uploaded_files)
         # print('list_docs ' ,list_docs)
         # print('list_save_path ' , list_save_path)
         passage_documents = util_get_list_page_and_passage(list_docs, list_save_path)
+        for passage_document in passage_documents:
+            text = passage_document.page_content
+            metadata = passage_document.metadata
+            preds = model(text)
+            my_list_structure.append({"text": text, "metadata": metadata,"preds":preds })
+
+    df = pd.DataFrame(my_list_structure)
+    df