Change placement of spaces decorator

app.py

@@ -73,6 +73,7 @@ def noisify_answer(input_ids, answer_start, threshold=1.0, eot_weight=1.0):
             noised[idx] = val
     return noised
 
+@spaces.GPU
 def generate_diffusion_text(input_ids, answer_start):
     with torch.no_grad():
         input_tensor = torch.tensor([input_ids], dtype=torch.long).to(model.device)
@@ -84,11 +85,13 @@ def generate_diffusion_text(input_ids, answer_start):
 
 # --- Inference Wrapper ---
 
-@spaces.GPU
+
 def diffusion_chat(question, eot_weight, max_it, sharpness):
     placeholder = "What do you know about the city of New York?"
     if question.strip() == "":
         question = placeholder
+    
+    print('started generation')
 
     prompt = f"User: {question}\nAssistant:"
     input_ids = tokenizer.encode(prompt, add_special_tokens=False)