Update app.py

app.py

@@ -115,7 +115,7 @@ def load_bnb_4bit_pipeline():
         print(f"4-bit BNB pipeline: {e}")
         raise
 
-# --- Image Generation and Shuffling Function ---
+@spaces.GPU(duration=240)
 def generate_images(prompt, quantization_choice, progress=gr.Progress(track_tqdm=True)):
     """Loads original and selected quantized model, generates one image each, clears memory, shuffles results."""
     if not prompt: