Messing with configuration.

app.py

@@ -47,7 +47,7 @@ print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
 # classifier = pipeline(task="zero-shot-classification", model=model, tokenizer=tokenizer, device="cuda:0")
 
 print(f"Testing 1")
-
+@spaces.GPU()
 def classify(data_string, request: gradio.Request):
     if request:
         if request.headers["origin"] not in ["https://statosphere-3704059fdd7e.c5v4v4jx6pq5.win", "https://crunchatize-77a78ffcc6a6.c5v4v4jx6pq5.win", "https://crunchatize-2-2b4f5b1479a6.c5v4v4jx6pq5.win", "https://tamabotchi-2dba63df3bf1.c5v4v4jx6pq5.win", "https://ravenok-statosphere-backend.hf.space", "https://lord-raven.github.io"]:
@@ -59,7 +59,6 @@ def classify(data_string, request: gradio.Request):
     return zero_shot_classification(data)
 
 print(f"Testing 2")
-# @spaces.GPU()
 def zero_shot_classification(data):
     results = []
     # classifier(data['sequence'], candidate_labels=data['candidate_labels'], hypothesis_template=data['hypothesis_template'], multi_label=data['multi_label'])