islasher commited on
Commit
2d2a8dc
·
verified ·
1 Parent(s): e2ee65c

Update ImageGenerator.py

Browse files
Files changed (1) hide show
  1. 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: