Update app.py

app.py

@@ -1,31 +1,27 @@
-import gradio as gr
 import torch
-from diffusers import DiffusionPipeline
+from diffusers import StableDiffusionPipeline
+import gradio as gr
 
-# Load the model with authentication
-pipe = DiffusionPipeline.from_pretrained("Shakker-Labs/AWPortrait-FL", torch_dtype=torch.float32, use_auth_token=True)
-pipe.to("cpu")   # Save some VRAM by offloading the model to CPU
+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
 
 def generate_image(prompt):
-    image = pipe(
-        prompt,
-        height=1024,
-        width=1024,
-        guidance_scale=3.5,
-        num_inference_steps=50,
-        max_sequence_length=512,
-        generator=torch.Generator("cpu").manual_seed(0)
-    ).images[0]
+    image = pipe(prompt).images[0]
+    return image
+def chatbot(prompt):
+    # Generate the image based on the user's input
+    image = generate_image(prompt)
     return image
 
-# Create the UI
-ui = gr.Interface(
-    generate_image,
-    gr.Textbox(lines=2, placeholder="Enter your prompt here..."),
-    "image",
-    title="FLUX.1-dev Image Generator",
-    description="Generate images using the FLUX.1-dev model.",
+# Create the Gradio interface
+interface = gr.Interface(
+    fn=chatbot,
+    inputs="text",
+    outputs="image",
+    title="RealVisXL V4.0 Text-to-Image Chatbot",
+    description="Enter a text prompt and get an AI-generated image."
 )
 
-# Launch the UI
-ui.launch()
+# Launch the interface
+interface.launch()