Update app.py

app.py

@@ -84,6 +84,7 @@ def clean_chat_history(chat_history):
             cleaned.append(msg)
     return cleaned
 
+@spaces.GPU
 def chat_generate(input_text: str, chat_history: list, max_new_tokens: int, temperature: float, top_p: float, top_k: int, repetition_penalty: float):
     """
     Chat generation using a text-only model.
@@ -241,6 +242,7 @@ def apply_style(style_name: str, positive: str, negative: str = ""):
         p, n = styles[DEFAULT_STYLE_NAME]
     return p.replace("{prompt}", positive), n + (negative if negative else "")
 
+@spaces.GPU
 def generate_image_lora(prompt: str, negative_prompt: str, use_negative_prompt: bool, seed: int, width: int, height: int, guidance_scale: float, randomize_seed: bool, style_name: str, lora_model: str):
     seed = int(randomize_seed_fn(seed, randomize_seed))
     positive_prompt, effective_negative_prompt = apply_style(style_name, prompt, negative_prompt)