Update app.py

app.py

@@ -1,46 +1,45 @@
 import gradio as gr
 import torch
-from diffusers import StableDiffusionPipeline
+from diffusers import StableDiffusionImg2ImgPipeline
 from PIL import Image
-import numpy as np
 
-# Lightweight SD pipeline (no MagicDrive complexity)
-pipe = StableDiffusionPipeline.from_pretrained(
+# Simple img2img pipeline (MagicDrive effect)
+pipe = StableDiffusionImg2ImgPipeline.from_pretrained(
     "runwayml/stable-diffusion-v1-5",
-    torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
+    torch_dtype=torch.float16,
     safety_checker=None
 )
 pipe = pipe.to("cuda" if torch.cuda.is_available() else "cpu")
 
-def generate_street_view(image, prompt, seed=42):
+def generate(image, prompt, strength=0.75, steps=20):
     if image is None:
-        return None, "Upload input image first"
+        return None, "Upload an image first"
     
-    # Simulate MagicDrive: SD img2img
-    torch.manual_seed(seed)
     result = pipe(
-        prompt, 
+        prompt=prompt,
         image=image,
-        strength=0.75,
-        num_inference_steps=20,
+        strength=strength,
+        num_inference_steps=steps,
         guidance_scale=7.5
     ).images[0]
     
-    return result, f"Street view generated! Seed: {seed}"
+    return result, f"Generated: {steps} steps, strength {strength}"
 
-with gr.Blocks(title="MagicDrive Demo") as demo:
-    gr.Markdown("# 🪄 MagicDrive-Style Street View Generator")
+with gr.Blocks() as demo:
+    gr.Markdown("# 🚗 Street View Generator (MagicDrive Style)")
     
     with gr.Row():
+        input_img = gr.Image(label="Input Image")
         with gr.Column():
-            input_img = gr.Image(type="pil", label="Input Street Image")
-            prompt = gr.Textbox("realistic street view, detailed cars, buildings", label="Prompt")
-            seed = gr.Slider(0, 10000, value=42, label="Seed")
+            prompt = gr.Textbox("realistic street view, detailed, 4k", label="Prompt")
+            strength = gr.Slider(0.5, 0.9, 0.75, label="Strength")
+            steps = gr.Slider(15, 50, 20, label="Steps")
             btn = gr.Button("Generate", variant="primary")
-        
-        output_img = gr.Image(label="Output Street View")
-        status = gr.Textbox(label="Status")
     
-    btn.click(generate_street_view, [input_img, prompt, seed], [output_img, status])
+    output_img = gr.Image(label="Output")
+    status = gr.Textbox(label="Status")
+    
+    btn.click(generate, [input_img, prompt, strength, steps], [output_img, status])
 
-demo.launch()
+if __name__ == "__main__":
+    demo.launch()