Jongmo commited on
Commit
814e455
1 Parent(s): c91a340

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,6 +5,7 @@ import Prove_lite as prv
5
  import pandas as pd
6
  import numpy as np
7
  import os
 
8
 
9
  def wtr_process(qid):
10
  try:
@@ -49,7 +50,6 @@ def wtr_process(qid):
49
  return error_df
50
 
51
  import spaces
52
- @spaces.GPU
53
  def prv_process(qid):
54
  target_QID = qid
55
  conn = sqlite3.connect('wikidata_claims_refs_parsed.db')
@@ -100,7 +100,7 @@ with gr.Blocks() as demo:
100
  )
101
  out_2 = gr.HTML(label="Results")
102
  run_button_2 = gr.Button("Start processing")
103
- run_button_2.click(prv_process, inp, out_2)
104
 
105
 
106
  if __name__ == "__main__":
 
5
  import pandas as pd
6
  import numpy as np
7
  import os
8
+ import spaces
9
 
10
  def wtr_process(qid):
11
  try:
 
50
  return error_df
51
 
52
  import spaces
 
53
  def prv_process(qid):
54
  target_QID = qid
55
  conn = sqlite3.connect('wikidata_claims_refs_parsed.db')
 
100
  )
101
  out_2 = gr.HTML(label="Results")
102
  run_button_2 = gr.Button("Start processing")
103
+ run_button_2.click(fn=spaces.GPU(prv_process), inp, out_2)
104
 
105
 
106
  if __name__ == "__main__":