Update app.py
Browse files
app.py
CHANGED
@@ -9,6 +9,8 @@ import time
|
|
9 |
class Dummy():
|
10 |
pass
|
11 |
|
|
|
|
|
12 |
# Ng
|
13 |
default_negative_prompt= "Logo,Watermark,Text,Ugly,Morbid,Extra fingers,Poorly drawn hands,Mutation,Blurry,Extra limbs,Gross proportions,Missing arms,Mutated hands,Long neck,Duplicate,Mutilated,Mutilated hands,Poorly drawn face,Deformed,Bad anatomy,Cloned face,Malformed limbs,Missing legs,Too many fingers"
|
14 |
|
@@ -54,10 +56,14 @@ def infer(prompt,negative_prompt,seed,resolution):
|
|
54 |
# generator = torch.Generator("cuda").manual_seed(555)
|
55 |
t=time.time()
|
56 |
|
57 |
-
if seed
|
58 |
generator=None
|
59 |
else:
|
60 |
-
|
|
|
|
|
|
|
|
|
61 |
|
62 |
w,h = resolution.split()
|
63 |
w,h = int(w),int(h)
|
@@ -90,7 +96,7 @@ with gr.Blocks(css=css) as demo:
|
|
90 |
with gr.Group():
|
91 |
with gr.Column():
|
92 |
prompt_in = gr.Textbox(label="Prompt", value="A red colored sports car")
|
93 |
-
resolution = gr.Dropdown(value=
|
94 |
seed = gr.Textbox(label="Seed", value=-1)
|
95 |
negative_prompt = gr.Textbox(label="Negative Prompt", value=default_negative_prompt)
|
96 |
submit_btn = gr.Button("Generate")
|
|
|
9 |
class Dummy():
|
10 |
pass
|
11 |
|
12 |
+
resolutions = ["1024 1024","1344 768","768 1344"]
|
13 |
+
|
14 |
# Ng
|
15 |
default_negative_prompt= "Logo,Watermark,Text,Ugly,Morbid,Extra fingers,Poorly drawn hands,Mutation,Blurry,Extra limbs,Gross proportions,Missing arms,Mutated hands,Long neck,Duplicate,Mutilated,Mutilated hands,Poorly drawn face,Deformed,Bad anatomy,Cloned face,Malformed limbs,Missing legs,Too many fingers"
|
16 |
|
|
|
56 |
# generator = torch.Generator("cuda").manual_seed(555)
|
57 |
t=time.time()
|
58 |
|
59 |
+
if seed=="-1":
|
60 |
generator=None
|
61 |
else:
|
62 |
+
try:
|
63 |
+
seed=int(seed)
|
64 |
+
generator = torch.Generator("cuda").manual_seed(seed)
|
65 |
+
except:
|
66 |
+
generator=None
|
67 |
|
68 |
w,h = resolution.split()
|
69 |
w,h = int(w),int(h)
|
|
|
96 |
with gr.Group():
|
97 |
with gr.Column():
|
98 |
prompt_in = gr.Textbox(label="Prompt", value="A red colored sports car")
|
99 |
+
resolution = gr.Dropdown(value=resolutions[0], show_label=True, label="Resolution", choices=resolutions)
|
100 |
seed = gr.Textbox(label="Seed", value=-1)
|
101 |
negative_prompt = gr.Textbox(label="Negative Prompt", value=default_negative_prompt)
|
102 |
submit_btn = gr.Button("Generate")
|