Update app.py
Browse files
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
|
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
|