Spaces:
Runtime error
Runtime error
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()
|