Update app.py

app.py

@@ -6,6 +6,7 @@ from longcat_image.models import LongCatImageTransformer2DModel
 from longcat_image.pipelines import LongCatImageEditPipeline
 import numpy as np
 import os
+import spaces
 
 # Global variables for model
 pipe = None
@@ -51,6 +52,7 @@ def initialize_model():
         print(f"❌ Error loading model: {e}")
         raise
 
+@spaces.GPU(duration=120)
 def edit_image(
     input_image: Image.Image,
     prompt: str,
@@ -81,7 +83,7 @@ def edit_image(
         progress(0.2, desc="Generating edited image...")
         
         # Set random seed for reproducibility
-        generator = torch.Generator("cpu").manual_seed(seed)
+        generator = torch.Generator("cuda" if torch.cuda.is_available() else "cpu").manual_seed(seed)
         
         # Run the pipeline
         with torch.inference_mode():
@@ -124,6 +126,9 @@ with gr.Blocks(fill_height=True) as demo:
             <p style="font-size: 14px; margin-top: 10px;">
                 Built with <a href="https://huggingface.co/spaces/akhaliq/anycoder" target="_blank" style="color: #4A90E2; text-decoration: none;">anycoder</a>
             </p>
+            <p style="font-size: 12px; color: #888; margin-top: 5px;">
+                ⚡ Powered by Zero-GPU
+            </p>
         </div>
     """)
     
@@ -179,6 +184,15 @@ with gr.Blocks(fill_height=True) as demo:
             
             edit_btn = gr.Button("✨ Edit Image", variant="primary", size="lg")
             
+            gr.Markdown("""
+            <div style="padding: 10px; background-color: #f0f7ff; border-radius: 8px; margin-top: 10px;">
+                <p style="margin: 0; font-size: 12px; color: #555;">
+                    ⏱️ <strong>Note:</strong> Zero-GPU provides 120 seconds of GPU time per request. 
+                    Processing typically takes 30-60 seconds depending on settings.
+                </p>
+            </div>
+            """)
+            
         with gr.Column(scale=1):
             gr.Markdown("### 🎯 Output")
             output_image = gr.Image(
@@ -195,6 +209,7 @@ with gr.Blocks(fill_height=True) as demo:
             - Supports both English and Chinese prompts
             - Try different guidance scales for varied results
             - Higher inference steps = better quality (but slower)
+            - GPU time is limited - optimize your settings for speed
             """)
     
     # Examples section
@@ -224,7 +239,7 @@ with gr.Blocks(fill_height=True) as demo:
     gr.HTML("""
         <div style="text-align: center; margin-top: 40px; padding: 20px; border-top: 1px solid #eee;">
             <p style="color: #666; font-size: 14px;">
-                Powered by LongCat Image Edit | 
+                Powered by LongCat Image Edit with Zero-GPU | 
                 <a href="https://huggingface.co/spaces/akhaliq/anycoder" target="_blank" style="color: #4A90E2;">Built with anycoder</a>
             </p>
         </div>