Spaces:
Sleeping
Sleeping
Update ImageGenerator.py
Browse files- ImageGenerator.py +1 -0
ImageGenerator.py
CHANGED
|
@@ -123,6 +123,7 @@ pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float
|
|
| 123 |
# pipe = pipe.to("cuda")
|
| 124 |
def generate_image(prompt):
|
| 125 |
# Get the part after the dash
|
|
|
|
| 126 |
if "-" in prompt:
|
| 127 |
prompt_to_send = prompt.split("-", 1)[-1].strip()
|
| 128 |
else:
|
|
|
|
| 123 |
# pipe = pipe.to("cuda")
|
| 124 |
def generate_image(prompt):
|
| 125 |
# Get the part after the dash
|
| 126 |
+
print('PROMPT TO INTRODUCE TO STABLDIFFUSION:', prompt)
|
| 127 |
if "-" in prompt:
|
| 128 |
prompt_to_send = prompt.split("-", 1)[-1].strip()
|
| 129 |
else:
|