RanM commited on
Commit
754c202
·
verified ·
1 Parent(s): 6dc4bbd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ model = AutoPipelineForText2Image.from_pretrained("stabilityai/sdxl-turbo")
10
  async def generate_image(prompt, prompt_name):
11
  try:
12
  print(f"Generating image for {prompt_name}")
13
- output = await (model, prompt=prompt, num_inference_steps=1, guidance_scale=0.0)
14
  image = output.images[0]
15
  img_bytes = image.tobytes()
16
  print(f"Image bytes length for {prompt_name}: {len(img_bytes)}")
 
10
  async def generate_image(prompt, prompt_name):
11
  try:
12
  print(f"Generating image for {prompt_name}")
13
+ output = await model(prompt=prompt, num_inference_steps=1, guidance_scale=0.0)
14
  image = output.images[0]
15
  img_bytes = image.tobytes()
16
  print(f"Image bytes length for {prompt_name}: {len(img_bytes)}")