Spaces:
Running
Running
Update app.py
Browse files
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=
|
196 |
seed=seed,
|
197 |
-
prompt = [generation.Prompt(text=prompt,parameters=generation.PromptParameters(weight=
|
198 |
-
generation.Prompt(text=negprompt,parameters=generation.PromptParameters(weight=-
|
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"
|