Update app.py

app.py

@@ -1,3 +1,4 @@
+import spaces
 import gradio as gr
 import torch
 from PIL import Image
@@ -6,7 +7,6 @@ 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
@@ -21,26 +21,27 @@ def initialize_model():
     
     try:
         device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-        checkpoint_dir = './weights/LongCat-Image-Edit'
+        model_id = 'meituan-longcat/LongCat-Image-Edit'
         
-        # Check if model exists
-        if not os.path.exists(checkpoint_dir):
-            raise ValueError(f"Model not found at {checkpoint_dir}. Please download the model first.")
+        print(f"🔄 Loading model from {model_id}...")
         
+        # Load text processor
         text_processor = AutoProcessor.from_pretrained(
-            checkpoint_dir, 
+            model_id, 
             subfolder='tokenizer'
         )
         
+        # Load transformer
         transformer = LongCatImageTransformer2DModel.from_pretrained(
-            checkpoint_dir, 
+            model_id, 
             subfolder='transformer',
             torch_dtype=torch.bfloat16, 
             use_safetensors=True
         ).to(device)
         
+        # Load pipeline
         pipe = LongCatImageEditPipeline.from_pretrained(
-            checkpoint_dir,
+            model_id,
             transformer=transformer,
             text_processor=text_processor,
         )
@@ -127,7 +128,7 @@ with gr.Blocks(fill_height=True) as demo:
                 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
+                ⚡ Powered by Zero-GPU | 🤗 Model: <a href="https://huggingface.co/meituan-longcat/LongCat-Image-Edit" target="_blank" style="color: #4A90E2;">meituan-longcat/LongCat-Image-Edit</a>
             </p>
         </div>
     """)
@@ -188,6 +189,7 @@ with gr.Blocks(fill_height=True) as demo:
             <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. 
+                    First run may take longer as the model loads from Hugging Face Hub.
                     Processing typically takes 30-60 seconds depending on settings.
                 </p>
             </div>
@@ -210,6 +212,7 @@ with gr.Blocks(fill_height=True) as demo:
             - Try different guidance scales for varied results
             - Higher inference steps = better quality (but slower)
             - GPU time is limited - optimize your settings for speed
+            - Model loads automatically from Hugging Face Hub
             """)
     
     # Examples section
@@ -239,7 +242,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 with Zero-GPU | 
+                Powered by <a href="https://huggingface.co/meituan-longcat/LongCat-Image-Edit" target="_blank" style="color: #4A90E2;">LongCat Image Edit</a> with Zero-GPU | 
                 <a href="https://huggingface.co/spaces/akhaliq/anycoder" target="_blank" style="color: #4A90E2;">Built with anycoder</a>
             </p>
         </div>