Update app.py

app.py

@@ -21,7 +21,6 @@ class ModelProcessor:
 
     
     @torch.inference_mode()
-    @spaces.GPU
     def process_data_and_compute_statistics(self, prompt):
         # Tokenize the prompt and move to the device
         tokens = self.tokenizer(
@@ -67,7 +66,7 @@ class ModelProcessor:
 
 processor = ModelProcessor()
 
-
+@spaces.GPU
 def detect(prompt):
     prediction = processor.process_data_and_compute_statistics(prompt)
     if prediction == 1: