fix

app.py

@@ -10,7 +10,7 @@ import gradio as gr
 import pandas as pd
 import polars as pl
 import matplotlib.pyplot as plt
-# import spaces
+import spaces
 from gradio_huggingfacehub_search import HuggingfaceHubSearch
 from huggingface_hub import PyTorchModelHubMixin
 import torch
@@ -50,7 +50,7 @@ model = QualityModel.from_pretrained("nvidia/quality-classifier-deberta").to(dev
 model.eval()
 
 
-# @spaces.GPU
+@spaces.GPU
 def predict(texts: list[str]):
     inputs = tokenizer(
         texts, return_tensors="pt", padding="longest", truncation=True
@@ -81,11 +81,7 @@ def plot_and_df(texts, preds):
         )
 
 
-# def download_data(dataset, config, split, column):
-#
-
-
-# @spaces.GPU
+@spaces.GPU
 def run_quality_check(dataset, config, split, column, batch_size, num_examples):
     logging.info(f"Fetching data for {dataset=} {config=} {split=} {column=}")
     try: