Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import numpy as np
 import random
+import spaces
 import torch
 import time
 import requests
@@ -32,6 +33,7 @@ pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=dtyp
 pipe.to(device)
 print("✅ Flux pipeline loaded.")
 
+@spaces.GPU
 def translate_albanian_to_english(text):
     """Translate Albanian to English using sepioo-facebook-translation API."""
     if not text.strip():
@@ -54,6 +56,7 @@ def translate_albanian_to_english(text):
                 return f"Përkthimi dështoi: {str(e)}. Provoni përsëri ose përdorni anglisht."
     return f"Përkthimi dështoi. Provoni përsëri ose përdorni anglisht."
 
+@spaces.GPU
 def generate_image(prompt: str, seed: int = 42, aspect_ratio: str = "16:9", randomize_seed: bool = False):
     """Generate image from English prompt."""
     if pipe is None: