Update app.py

Replace with text-to-image pipeline

app.py

@@ -5,7 +5,11 @@ import gradio as gr
 import requests
 import spaces
 import torch
-from diffusers import AutoencoderKL, StableDiffusionXLImg2ImgPipeline
+from diffusers import (
+    AutoencoderKL,
+    StableDiffusionXLImg2ImgPipeline,
+    StableDiffusionXLPipeline,
+)
 from loguru import logger
 from PIL import Image
 from tqdm import tqdm
@@ -44,7 +48,8 @@ download(
 
 vae = AutoencoderKL.from_single_file(vae_path)
 
-pipe = StableDiffusionXLImg2ImgPipeline.from_single_file(
+# pipe = StableDiffusionXLImg2ImgPipeline.from_single_file(
+pipe = StableDiffusionXLPipeline.from_single_file(
     model_path, torch_dtype=torch.float16, use_safetensors=True, variant="fp16", vae=vae
 )
 pipe = pipe.to("cuda")
@@ -54,18 +59,16 @@ pipe = pipe.to("cuda")
 @spaces.GPU
 def generate(
     prompt: str,
-    init_image: Image.Image,
+    # init_image: Image.Image,
     strength: float,
     num_inference_steps: int,
     guidance_scale: float,
     progress=gr.Progress(track_tqdm=True),
 ):
-    logger.info(
-        f"Starting image generation: {dict(prompt=prompt, image=init_image, strength=strength)}"
-    )
+    logger.info(f"Starting image generation: {dict(prompt=prompt, strength=strength)}")
 
-    # Downscale the image
-    init_image.thumbnail((1024, 1024))
+    # # Downscale the image
+    # init_image.thumbnail((1024, 1024))
 
     additional_args = {
         k: v
@@ -79,7 +82,7 @@ def generate(
 
     images = pipe(
         prompt=prompt,
-        image=init_image,
+        # image=init_image,
         **additional_args,
     ).images
 
@@ -90,7 +93,7 @@ demo = gr.Interface(
     fn=generate,
     inputs=[
         gr.Text(label="Prompt"),
-        gr.Image(label="Init image", type="pil"),
+        # gr.Image(label="Init image", type="pil"),
         gr.Slider(label="Strength", minimum=0.0, maximum=1.0, value=0.0),
         gr.Slider(label="Number of inference steps", minimum=0, maximum=100, value=0),
         gr.Slider(label="Guidance scale", minimum=0.0, maximum=100.0, value=0.0),