K00B404 commited on
Commit
362000d
·
verified ·
1 Parent(s): 664d312

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -64,7 +64,7 @@ async def generate_image(prompt, model, lora_word, width, height, scales, steps,
64
  print(f"[-] Generating image with prompt: {text}, model: {model}")
65
  client = AsyncInferenceClient()
66
  image = await client.text_to_image(prompt=text, height=height, width=width, guidance_scale=scales, num_inference_steps=steps, model=model)
67
- return remove(image), seed
68
  except Exception as e:
69
  print(f"[-] Error generating image: {e}")
70
  return None, None
 
64
  print(f"[-] Generating image with prompt: {text}, model: {model}")
65
  client = AsyncInferenceClient()
66
  image = await client.text_to_image(prompt=text, height=height, width=width, guidance_scale=scales, num_inference_steps=steps, model=model)
67
+ return image, seed
68
  except Exception as e:
69
  print(f"[-] Error generating image: {e}")
70
  return None, None