Update app.py

app.py

@@ -20,7 +20,7 @@ MODELS = {
 DEFAULT_MODEL = "Spestly/Athena-R3X-8B"
 
 # GPU-accelerated function
-@space.GPU
+@spaces.GPU
 def load_model(model_name):
     model_id = MODELS.get(model_name, DEFAULT_MODEL)
     
@@ -42,7 +42,7 @@ def load_model(model_name):
     
     return model, tokenizer
 
-@space.GPU
+@spaces.GPU
 def generate_text(prompt, model_name, max_length=512, temperature=0.7):
     try:
         model, tokenizer = load_model(model_name)