Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import torch
 from transformers import AutoModelForCausalLM
 import os
+import spaces
 
 # Load the model
 model_id = "tencent/HunyuanImage-3.0"
@@ -11,6 +12,7 @@ print("Note: This is a very large model (80B params) and requires significant GP
 print("For production use, consider using the FAL API or other inference providers.")
 
 # For demo purposes, we'll use inference API
+@spaces.GPU
 def generate_image(prompt, seed=42, diff_infer_steps=50, image_size="auto"):
     """
     Generate image using HunyuanImage-3.0