Update app.py

app.py

@@ -9,12 +9,12 @@ model_id = "xmadai/Mistral-Large-Instruct-2407-xMADai-INT4"
 # Load the tokenizer
 tokenizer = AutoTokenizer.from_pretrained(model_id, use_fast=False, trust_remote_code=True)
 
-@spaces.DynamicGpuSupport()
+# Removed the invalid decorator
 class ModelWrapper:
     def __init__(self):
         self.model = None  # Model will be loaded when GPU is allocated
 
-    @spaces.GPU
+    @spaces.GPU  # Use the correct decorator
     def generate(self, prompt):
         if self.model is None:
             # Load the model when GPU is allocated