Update app.py

app.py

@@ -1,10 +1,50 @@
 import gradio as gr
+from diffusers import StableDiffusionImg2ImgPipeline
+import torch
+from PIL import Image
 
-with gr.Blocks(fill_height=True) as demo:
-    with gr.Sidebar():
-        gr.Markdown("# Inference Provider")
-        gr.Markdown("This Space showcases the nitrosocke/Ghibli-Diffusion model, served by the hf-inference API. Sign in with your Hugging Face account to use this API.")
-        button = gr.LoginButton("Sign in")
-    gr.load("models/nitrosocke/Ghibli-Diffusion", accept_token=button, provider="hf-inference")
+# Load the model (this runs when the Space starts)
+model_id = "nitrosocke/Ghibli-Diffusion"
+pipe = StableDiffusionImg2ImgPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
+pipe = pipe.to("cuda")  # Use GPU if available in the Space
+
+# Define the inference function
+def ghibli_transform(input_image, prompt="ghibli style", strength=0.75, guidance_scale=7.5):
+    # Convert Gradio image (numpy array) to PIL Image and resize
+    init_image = Image.fromarray(input_image).convert("RGB").resize((768, 768))
+    
+    # Generate the Ghibli-style image
+    output = pipe(
+        prompt=prompt,
+        init_image=init_image,
+        strength=strength,
+        guidance_scale=guidance_scale,
+        num_inference_steps=50
+    ).images[0]
+    
+    return output
+
+# Create the Gradio interface
+with gr.Blocks(title="Ghibli Diffusion Image Transformer") as demo:
+    gr.Markdown("# Ghibli Diffusion Image Transformer")
+    gr.Markdown("Upload an image and transform it into Studio Ghibli style using nitrosocke/Ghibli-Diffusion!")
+    
+    with gr.Row():
+        with gr.Column():
+            input_img = gr.Image(label="Upload Image", type="numpy")
+            prompt = gr.Textbox(label="Prompt", value="ghibli style")
+            strength = gr.Slider(0, 1, value=0.75, step=0.05, label="Strength (How much to transform)")
+            guidance = gr.Slider(1, 20, value=7.5, step=0.5, label="Guidance Scale")
+            submit_btn = gr.Button("Transform")
+        with gr.Column():
+            output_img = gr.Image(label="Ghibli-Style Output")
     
+    # Connect the button to the function
+    submit_btn.click(
+        fn=ghibli_transform,
+        inputs=[input_img, prompt, strength, guidance],
+        outputs=output_img
+    )
+
+# Launch the Space
 demo.launch()