Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
addb999
1
Parent(s):
44cbba4
Update app.py
Browse fileswrap run_quality_check with gpu decorator
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()
|