Update app.py
Browse files
app.py
CHANGED
|
@@ -73,7 +73,7 @@ def save_images(image_array, profile: gr.OAuthProfile | None, metadata: dict):
|
|
| 73 |
|
| 74 |
def generate(
|
| 75 |
prompt: str,
|
| 76 |
-
test: str
|
| 77 |
seed: int = 0,
|
| 78 |
guidance_scale: float = 8.0,
|
| 79 |
num_inference_steps: int = 4,
|
|
@@ -125,6 +125,13 @@ with gr.Blocks(css="style.css") as demo:
|
|
| 125 |
placeholder="Enter your prompt",
|
| 126 |
container=False,
|
| 127 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 128 |
run_button = gr.Button("Run", scale=0)
|
| 129 |
result = gr.Gallery(
|
| 130 |
label="Generated images", show_label=False, elem_id="gallery", grid=[2]
|
|
@@ -171,12 +178,13 @@ with gr.Blocks(css="style.css") as demo:
|
|
| 171 |
fn=generate,
|
| 172 |
inputs=[
|
| 173 |
prompt,
|
|
|
|
| 174 |
seed,
|
| 175 |
guidance_scale,
|
| 176 |
num_inference_steps,
|
| 177 |
randomize_seed
|
| 178 |
],
|
| 179 |
-
outputs=[result, seed,
|
| 180 |
api_name="run",
|
| 181 |
)
|
| 182 |
|
|
|
|
| 73 |
|
| 74 |
def generate(
|
| 75 |
prompt: str,
|
| 76 |
+
test: str,
|
| 77 |
seed: int = 0,
|
| 78 |
guidance_scale: float = 8.0,
|
| 79 |
num_inference_steps: int = 4,
|
|
|
|
| 125 |
placeholder="Enter your prompt",
|
| 126 |
container=False,
|
| 127 |
)
|
| 128 |
+
test = gr.Text(
|
| 129 |
+
label="test",
|
| 130 |
+
show_label=False,
|
| 131 |
+
max_lines=1,
|
| 132 |
+
placeholder="Enter your prompt",
|
| 133 |
+
container=False,
|
| 134 |
+
)
|
| 135 |
run_button = gr.Button("Run", scale=0)
|
| 136 |
result = gr.Gallery(
|
| 137 |
label="Generated images", show_label=False, elem_id="gallery", grid=[2]
|
|
|
|
| 178 |
fn=generate,
|
| 179 |
inputs=[
|
| 180 |
prompt,
|
| 181 |
+
test,
|
| 182 |
seed,
|
| 183 |
guidance_scale,
|
| 184 |
num_inference_steps,
|
| 185 |
randomize_seed
|
| 186 |
],
|
| 187 |
+
outputs=[result, seed, test],
|
| 188 |
api_name="run",
|
| 189 |
)
|
| 190 |
|