aifeifei798 commited on
Commit
87bc19f
·
verified ·
1 Parent(s): 3b7620a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -38,9 +38,8 @@ def infer(prompt, styles_Radio="(None)", seed=42, randomize_seed=False, width=10
38
  seed = random.randint(0, MAX_SEED)
39
  generator = torch.Generator().manual_seed(seed)
40
  if styles_Radio:
41
- style_name = styles_Radio
42
  for style in config.style_list:
43
- if style["name"] == style_name:
44
  prompt = style["prompt"].replace("{prompt}", prompt)
45
  image = pipe(
46
  prompt = "flux, 8k, ",
 
38
  seed = random.randint(0, MAX_SEED)
39
  generator = torch.Generator().manual_seed(seed)
40
  if styles_Radio:
 
41
  for style in config.style_list:
42
+ if style["name"] == styles_Radio:
43
  prompt = style["prompt"].replace("{prompt}", prompt)
44
  image = pipe(
45
  prompt = "flux, 8k, ",