Update app.py

app.py

@@ -2,12 +2,13 @@ import torch
 from diffusers import StableDiffusionPipeline
 import gradio as gr
 
-model_id = "SG161222/RealVisXL_V4.0"
-pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
-pipe.to("cpu")  # Use "cuda" if GPU is available
+# Use the CompVis stable-diffusion-v1-4 model
+model_id = "CompVis/stable-diffusion-v1-4"
+pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32)  # float32 for CPU
+pipe.to("cpu")  # Ensure it runs on CPU
 
 def generate_image(prompt, pipe):
-    # Generate the image using the pipeline with proper args handling
+    # Generate the image using the pipeline
     image = pipe(prompt, guidance_scale=7.5, num_inference_steps=50).images[0]
     return image
 
@@ -21,7 +22,7 @@ interface = gr.Interface(
     fn=chatbot,
     inputs="text",
     outputs="image",
-    title="RealVisXL V4.0 Text-to-Image Chatbot",
+    title="Stable Diffusion v1-4 Text-to-Image Chatbot",
     description="Enter a text prompt and get an AI-generated image."
 )