Update app.py

app.py

@@ -1,44 +1,34 @@
-import subprocess
-import sys
-
-def install(package):
-    subprocess.check_call([sys.executable, "-m", "pip", "install", package])
-
-# Attempt to import transformers and install if missing
-try:
-    import transformers
-except ModuleNotFoundError:
-    install("transformers")
-    import transformers
-
-# Then proceed with your imports
-from diffusers import StableDiffusionImg2ImgPipeline
 import gradio as gr
 import torch
+from diffusers import StableDiffusionImg2ImgPipeline
 from PIL import Image
 
-# Load the fine-tuned Studio Ghibli style model
-model_id = "nitrosocke/Ghibli-Diffusion"  # Fine-tuned model for Ghibli style
-pipe = StableDiffusionImg2ImgPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
-# Use GPU if available, otherwise fall back to CPU
+# Load the pre-trained Studio Ghibli style model
+model_id = "nitrosocke/Ghibli-Diffusion"
 device = "cuda" if torch.cuda.is_available() else "cpu"
-pipe.to(device)
+
+pipe = StableDiffusionImg2ImgPipeline.from_pretrained(
+    model_id, torch_dtype=torch.float16
+).to(device)
 
 def transform_image(input_image: Image.Image) -> Image.Image:
-    # Resize input image to 512x512 for consistency
     input_image = input_image.resize((512, 512))
     prompt = "ghibli style"
-    # 'strength' controls how much the input image is modified (0.0 to 1.0)
-    output = pipe(prompt=prompt, image=input_image, strength=0.75, guidance_scale=7.5)
+    output = pipe(
+        prompt=prompt,
+        image=input_image,
+        strength=0.75,
+        guidance_scale=7.5,
+        num_inference_steps=50,
+    )
     return output.images[0]
 
-# Create Gradio interface: input is an image, output is the transformed image.
 demo = gr.Interface(
     fn=transform_image,
     inputs=gr.Image(type="pil", label="Upload your portrait/photo"),
     outputs=gr.Image(type="pil", label="Studio Ghibli Style Output"),
     title="Studio Ghibli Style Converter",
-    description="Upload a portrait or photo and transform it into a Studio Ghibli-style image using a fine-tuned Stable Diffusion model."
+    description="Upload a portrait or photo to transform it into a Studio Ghibli-style image.",
 )
 
 if __name__ == "__main__":