Update app.py

app.py

@@ -6,14 +6,14 @@ import torch
 
 
 class ModelProcessor:
-    device = "cuda:0" if torch.cuda.is_available() else "cpu"
     def __init__(self, repo_id="HuggingFaceTB/cosmo-1b"):
+        self.device = "cuda:0" if torch.cuda.is_available() else "cpu"
         # Initialize the tokenizer
         self.tokenizer = AutoTokenizer.from_pretrained(repo_id, use_fast=True)
 
         # Initialize and configure the model
         self.model = AutoModelForCausalLM.from_pretrained(
-            repo_id, torch_dtype=torch.float16, device_map={"": device}, trust_remote_code=True
+            repo_id, torch_dtype=torch.float16, device_map={"": self.device}, trust_remote_code=True
         )
         self.model.eval()  # Set the model to evaluation mode
 
@@ -67,7 +67,7 @@ class ModelProcessor:
 
 processor = ModelProcessor()
 
-@spaces.GPU
+@spaces.GPU(duration=180)
 def detect(prompt):
     prediction = processor.process_data_and_compute_statistics(prompt)
     if prediction == 1: