Update app.py

app.py

@@ -1,11 +1,9 @@
 import gradio as gr
 from diffusers import StableDiffusionPipeline
-import torch
 
 # Load pre-trained model
 model_id = "CompVis/stable-diffusion-v1-4"
-pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
-pipe = pipe.to("cpu")  # 'cuda' 대신 'cpu'로 변경
+pipe = StableDiffusionPipeline.from_pretrained(model_id)
 
 def generate_image(prompt):
     # Generate image from text prompt
@@ -15,8 +13,8 @@ def generate_image(prompt):
 # Define Gradio UI
 iface = gr.Interface(
     fn=generate_image,
-    inputs=gr.inputs.Textbox(lines=2, placeholder="Enter a text prompt here..."),
-    outputs="image",
+    inputs=gr.components.Textbox(lines=2, placeholder="Enter a text prompt here..."),
+    outputs=gr.components.Image(),
     title="DreamBooth Image Generation",
     description="Enter a text prompt to generate an image using DreamBooth.",
     examples=[