Update app.py

app.py

@@ -1,10 +1,10 @@
 import gradio as gr
-from diffusers import DiffusionPipeline
 import torch
 from gradio_client import Client
 pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16, use_safetensors=True, variant="fp16")
 if torch.cuda.is_available(): pipe.to("cuda")
 client = Client("stabilityai/triposr")
+sdxl = Client("hysts/SD-XL")
 
 
 ABOUT_TEXT = """
@@ -16,7 +16,24 @@ For image-to-3D, use [TripoSR](https://huggingface.co/spaces/stabilityai/TripoSR
 """.strip()
 def generate(text):
     # generate image
-    image = pipe(prompt=text).images[0]
+    image = sdxl.predict(
+		text,
+		"",
+		"",
+		"",
+		False,
+		False,
+		False,
+        0,
+		1024,
+		1024,
+		5,
+		5,
+		25,
+		25,
+		False,	# bool  in 'Apply refiner' Checkbox component
+		api_name="/run"
+    )
     # preprocess
     result = client.predict(
 		image,