Create app.py

app.py

@@ -0,0 +1,86 @@
+import gradio as gr
+import torch
+from PIL import Image
+from diffusers import QwenImageEditPipeline
+import os
+
+# Initialize model
+model_id = "Qwen/Qwen-Image-Edit"
+pipe = None
+
+def load_model():
+    global pipe
+    if pipe is None:
+        try:
+            pipe = QwenImageEditPipeline.from_pretrained(model_id, torch_dtype=torch.float32)
+            # For CPU compatibility, remove CUDA-specific optimizations
+            # Note: This will be slower but functional on CPU
+            pipe = pipe.to("cpu")
+            print("Model loaded successfully.")
+        except Exception as e:
+            print(f"Error loading model: {e}")
+            raise gr.Error("Failed to load model. Please check logs.")
+
+# Load model on startup
+load_model()
+
+def process_image(image, prompt):
+    if pipe is None:
+        raise gr.Error("Model not loaded.")
+    
+    # Ensure image is RGB
+    if image.mode != "RGB":
+        image = image.convert("RGB")
+
+    # Prepare inputs
+    inputs = {
+        "image": image,
+        "prompt": prompt,
+        "generator": torch.manual_seed(0),
+        "true_cfg_scale": 4.0,
+        "negative_prompt": " ",
+        "num_inference_steps": 50,
+    }
+
+    # Run inference
+    with torch.inference_mode():
+        output = pipe(**inputs)
+        output_image = output.images[0]
+
+    return output_image
+
+# Gradio Interface
+with gr.Blocks(title="Qwen Image Editor", theme=gr.themes.Default(primary_hue="#615ced")) as demo:
+    gr.Markdown("# 🎨 Qwen Image Editor")
+    gr.Image(value="https://qianwen-res.oss-cn-beijing.aliyuncs.com/Qwen-Image/qwen_image_edit_logo.png", label="", height=80)
+    gr.Markdown("Edit images using text prompts powered by Qwen-Image-Edit")
+    
+    with gr.Row():
+        with gr.Column():
+            image_input = gr.Image(type="pil", label="Input Image")
+            prompt_input = gr.Textbox(label="Edit Prompt", placeholder="Example: Change the rabbit's color to purple...")
+            submit_btn = gr.Button("Generate Edited Image", variant="primary")
+        
+        with gr.Column():
+            image_output = gr.Image(label="Output Image", interactive=False)
+
+    submit_btn.click(
+        fn=process_image,
+        inputs=[image_input, prompt_input],
+        outputs=image_output
+    )
+
+    gr.Examples(
+        examples=[
+            ["https://qianwen-res.oss-cn-beijing.aliyuncs.com/Qwen-Image/edit_en/%E5%B9%BB%E7%81%AF%E7%89%873.JPG", "Make the capybara wear sunglasses"],
+            ["https://qianwen-res.oss-cn-beijing.aliyuncs.com/Qwen-Image/edit_en/%E5%B9%BB%E7%81%AF%E7%89%8712.JPG", "Rotate the object 90 degrees clockwise"],
+            ["https://qianwen-res.oss-cn-beijing.aliyuncs.com/Qwen-Image/edit_en/%E5%B9%BB%E7%81%AF%E7%89%8715.JPG", "Change the text to 'Hello World' in blue"],
+        ],
+        inputs=[image_input, prompt_input],
+        outputs=image_output,
+        fn=process_image,
+        cache_examples=False
+    )
+
+if __name__ == "__main__":
+    demo.launch()