Update app.py from anycoder

app.py

@@ -1,13 +1,14 @@
 from PIL import Image
 import torch
 import gradio as gr
+import spaces
 
 # Load models
 model2 = torch.hub.load(
     "AK391/animegan2-pytorch:main",
     "generator",
     pretrained=True,
-    device="cuda",
+    device="cpu",  # Load on CPU initially
     progress=False
 )
 
@@ -15,23 +16,29 @@ model1 = torch.hub.load(
     "AK391/animegan2-pytorch:main", 
     "generator", 
     pretrained="face_paint_512_v1",  
-    device="cuda"
+    device="cpu"  # Load on CPU initially
 )
 
 face2paint = torch.hub.load(
     'AK391/animegan2-pytorch:main', 
     'face2paint', 
     size=512, 
-    device="cuda",
+    device="cpu",  # Load on CPU initially
     side_by_side=False
 )
 
+@spaces.GPU  # Zero GPU decorator - moves to GPU only when function runs
 def inference(img, ver):
     """Convert portrait to anime style"""
+    # Move models to GPU when function is called
     if ver == 'Version 2':
+        model2.to('cuda')
         out = face2paint(model2, img)
+        model2.to('cpu')  # Move back to CPU to free GPU
     else:
+        model1.to('cuda')
         out = face2paint(model1, img)
+        model1.to('cpu')  # Move back to CPU to free GPU
     return out
 
 # Custom CSS for modern, mobile-friendly design
@@ -151,7 +158,7 @@ with gr.Blocks(fill_height=True) as demo:
     # Header
     with gr.Column(elem_id="header"):
         gr.Markdown("# ⚡ AnimeGAN v2")
-        gr.Markdown("Transform your portraits into stunning anime-style artwork")
+        gr.Markdown("Transform your portraits into stunning anime-style artwork using Zero GPU")
     
     # Main content
     with gr.Column(elem_id="main-content"):
@@ -232,6 +239,9 @@ with gr.Blocks(fill_height=True) as demo:
             <p style='margin-top: 1rem; font-size: 0.9rem; color: rgba(255,255,255,0.6);'>
                 💡 <strong>Tip:</strong> Use a cropped portrait photo for best results
             </p>
+            <p style='margin-top: 0.5rem; font-size: 0.85rem; color: rgba(255,255,255,0.5);'>
+                ⚡ Powered by Zero GPU - efficient GPU usage on Hugging Face Spaces
+            </p>
             """
         )