Update app.py

app.py

@@ -49,7 +49,7 @@ def wtr_process(qid):
             error_df = pd.DataFrame({'Error': [str(e)]})
             return error_df
         
-import spaces
+
 def prv_process(qid):
     target_QID = qid
     conn = sqlite3.connect('wikidata_claims_refs_parsed.db')
@@ -59,7 +59,7 @@ def prv_process(qid):
     query = f"SELECT * FROM html_text Where  entity_id = '{target_QID}'"
     reference_text_df = pd.read_sql_query(query, conn)
     
-    verbalised_claims_df_final = spaces.GPU(prv.verbalisation(claim_df))
+    verbalised_claims_df_final = prv.verbalisation(claim_df)
 
     progress = gr.Progress(len(verbalised_claims_df_final))  # Create progress bar for Gradio
     def update_progress(curr_step, total_steps):
@@ -100,7 +100,7 @@ with gr.Blocks() as demo:
     )
     out_2 = gr.HTML(label="Results")
     run_button_2 = gr.Button("Start processing")
-    run_button_2.click(fn=prv_process, inp, out_2)
+    run_button_2.click(fn=spaces.GPU(prv_process), inp, out_2)
 
     
 if __name__ == "__main__":