greendra commited on
Commit
0babc75
·
1 Parent(s): c6b3a2f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -192,10 +192,10 @@ def sd(prompt, negprompt):
192
  # set random seed
193
  answers = stability_api.generate(
194
  #prompt=prompt,
195
- steps=10,
196
  seed=seed,
197
- prompt = [generation.Prompt(text=prompt,parameters=generation.PromptParameters(weight=1)),
198
- generation.Prompt(text=negprompt,parameters=generation.PromptParameters(weight=-1))],
199
  )
200
 
201
  # convert seed to string
@@ -219,7 +219,7 @@ def sd(prompt, negprompt):
219
  img = Image.open(io.BytesIO(artifact.binary))
220
  link = freeimagehost(prompt, img, seed)
221
  # time.sleep(2)
222
- print(keyid + ": " + timestamp + ": PROMPT \"" + prompt + "\" WAS GENERATED SUCCESSFULLY AND IMAGE LINK: \n" + link[1] + "\n")
223
  return img, 'Your download URL: ' + link[0] + '<br> Seed: ' + seed
224
  except Exception as e:
225
  print(keyid, '\n' ,e)
@@ -227,7 +227,7 @@ def sd(prompt, negprompt):
227
  sdtries += 1
228
  if sdtries < 5:
229
  print("Retrying...")
230
- return sd(prompt)
231
  else:
232
  sdtries = 0
233
  return Image.open('error.png'), "Error! Please try again"
 
192
  # set random seed
193
  answers = stability_api.generate(
194
  #prompt=prompt,
195
+ steps=15,
196
  seed=seed,
197
+ prompt = [generation.Prompt(text=prompt,parameters=generation.PromptParameters(weight=5)),
198
+ generation.Prompt(text=negprompt,parameters=generation.PromptParameters(weight=-5))],
199
  )
200
 
201
  # convert seed to string
 
219
  img = Image.open(io.BytesIO(artifact.binary))
220
  link = freeimagehost(prompt, img, seed)
221
  # time.sleep(2)
222
+ print(keyid + ": " + timestamp + ": PROMPT \"" + prompt + ", Neg: " + negprompt + "\" WAS GENERATED SUCCESSFULLY AND IMAGE LINK: \n" + link[1] + "\n")
223
  return img, 'Your download URL: ' + link[0] + '<br> Seed: ' + seed
224
  except Exception as e:
225
  print(keyid, '\n' ,e)
 
227
  sdtries += 1
228
  if sdtries < 5:
229
  print("Retrying...")
230
+ return sd(prompt, negprompt)
231
  else:
232
  sdtries = 0
233
  return Image.open('error.png'), "Error! Please try again"