polinaeterna HF staff commited on
Commit
addb999
1 Parent(s): 44cbba4

Update app.py

Browse files

wrap run_quality_check with gpu decorator

Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -72,6 +72,7 @@ def plot_and_df(texts, preds):
72
  )
73
 
74
 
 
75
  def run_quality_check(dataset, column, batch_size, num_examples):
76
  # config = "default"
77
  info_resp = session.get(f"https://datasets-server.huggingface.co/info?dataset={dataset}", timeout=3).json()
 
72
  )
73
 
74
 
75
+ @spaces.GPU
76
  def run_quality_check(dataset, column, batch_size, num_examples):
77
  # config = "default"
78
  info_resp = session.get(f"https://datasets-server.huggingface.co/info?dataset={dataset}", timeout=3).json()