Update app.py

app.py

@@ -99,7 +99,7 @@ def app2():
         submit_button = st.form_submit_button(label="🍃 Get me the data!")
       
     if len(doc)!=0:
-        pred_df = ner_prediction(corpus=doc, compute='gpu') #pass compute='gpu' if using gpu
+        pred_df = ner_prediction(corpus=doc, compute='cpu') #pass compute='gpu' if using gpu
         with c3:
             st.dataframe(pred_df)
             CSVButton1 = download_button(pred_df, "key-value-content.csv", "📥 Download (.csv)")
@@ -127,7 +127,7 @@ def app2():
                 final_df = pd.DataFrame(columns= ["Page","Entity Group","Value","Score"])
                 while page <  document.pageCount:
                     page_text=document.get_page_text(page)
-                    out = ner_prediction(corpus=page_text, compute='gpu')
+                    out = ner_prediction(corpus=page_text, compute='cpu')
                     output = out.drop_duplicates(subset=["value"],keep='first')
                     #to iterate through every row in the dataframe
                     for index, row in output.iterrows():