Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -1270,8 +1270,140 @@ def create_ui():
|
|
1270 |
with gr.Row():
|
1271 |
submit_tts = gr.Button(value="Submit", interactive=True)
|
1272 |
clear_tts = gr.Button(value="Clear", interactive=True)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1273 |
|
1274 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1275 |
|
1276 |
###############################################################################
|
1277 |
# this part is for 3d generate.
|
@@ -1382,133 +1514,7 @@ def create_ui():
|
|
1382 |
# above part is for 3d generate.
|
1383 |
###############################################################################
|
1384 |
|
1385 |
-
###############################################################################
|
1386 |
-
############# this part is for text to image #############
|
1387 |
-
###############################################################################
|
1388 |
-
|
1389 |
-
with gr.Row(variant="panel") as text2image_model:
|
1390 |
-
with gr.Row():
|
1391 |
-
prompt = gr.Text(
|
1392 |
-
label="Prompt",
|
1393 |
-
show_label=False,
|
1394 |
-
max_lines=1,
|
1395 |
-
placeholder="Enter your prompt",
|
1396 |
-
container=False,
|
1397 |
-
)
|
1398 |
-
run_button = gr.Button("Run", scale=0)
|
1399 |
-
result = gr.Gallery(label="Result", columns=1, show_label=False)
|
1400 |
-
with gr.Accordion("Advanced options", open=False):
|
1401 |
-
num_images = gr.Slider(
|
1402 |
-
label="Number of Images",
|
1403 |
-
minimum=1,
|
1404 |
-
maximum=4,
|
1405 |
-
step=1,
|
1406 |
-
value=1,
|
1407 |
-
)
|
1408 |
-
with gr.Row():
|
1409 |
-
use_negative_prompt = gr.Checkbox(label="Use negative prompt", value=True)
|
1410 |
-
negative_prompt = gr.Text(
|
1411 |
-
label="Negative prompt",
|
1412 |
-
max_lines=5,
|
1413 |
-
lines=4,
|
1414 |
-
placeholder="Enter a negative prompt",
|
1415 |
-
value="(deformed, distorted, disfigured:1.3), poorly drawn, bad anatomy, wrong anatomy, extra limb, missing limb, floating limbs, (mutated hands and fingers:1.4), disconnected limbs, mutation, mutated, ugly, disgusting, blurry, amputation, NSFW",
|
1416 |
-
visible=True,
|
1417 |
-
)
|
1418 |
-
seed = gr.Slider(
|
1419 |
-
label="Seed",
|
1420 |
-
minimum=0,
|
1421 |
-
maximum=MAX_SEED,
|
1422 |
-
step=1,
|
1423 |
-
value=0,
|
1424 |
-
)
|
1425 |
-
randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
|
1426 |
-
with gr.Row(visible=True):
|
1427 |
-
width = gr.Slider(
|
1428 |
-
label="Width",
|
1429 |
-
minimum=512,
|
1430 |
-
maximum=MAX_IMAGE_SIZE,
|
1431 |
-
step=64,
|
1432 |
-
value=1024,
|
1433 |
-
)
|
1434 |
-
height = gr.Slider(
|
1435 |
-
label="Height",
|
1436 |
-
minimum=512,
|
1437 |
-
maximum=MAX_IMAGE_SIZE,
|
1438 |
-
step=64,
|
1439 |
-
value=1024,
|
1440 |
-
)
|
1441 |
-
with gr.Row():
|
1442 |
-
guidance_scale = gr.Slider(
|
1443 |
-
label="Guidance Scale",
|
1444 |
-
minimum=0.1,
|
1445 |
-
maximum=6,
|
1446 |
-
step=0.1,
|
1447 |
-
value=3.0,
|
1448 |
-
)
|
1449 |
-
num_inference_steps = gr.Slider(
|
1450 |
-
label="Number of inference steps",
|
1451 |
-
minimum=1,
|
1452 |
-
maximum=15,
|
1453 |
-
step=1,
|
1454 |
-
value=8,
|
1455 |
-
)
|
1456 |
-
|
1457 |
-
gr.Examples(
|
1458 |
-
examples=examples,
|
1459 |
-
inputs=prompt,
|
1460 |
-
cache_examples=False
|
1461 |
-
)
|
1462 |
-
|
1463 |
-
use_negative_prompt.change(
|
1464 |
-
fn=lambda x: gr.update(visible=x),
|
1465 |
-
inputs=use_negative_prompt,
|
1466 |
-
outputs=negative_prompt,
|
1467 |
-
api_name=False,
|
1468 |
-
)
|
1469 |
-
|
1470 |
-
# gr.on(
|
1471 |
-
# triggers=[
|
1472 |
-
# prompt.submit,
|
1473 |
-
# negative_prompt.submit,
|
1474 |
-
# run_button.click,
|
1475 |
-
# ],
|
1476 |
-
# fn=generate,
|
1477 |
-
# inputs=[
|
1478 |
-
# prompt,
|
1479 |
-
# negative_prompt,
|
1480 |
-
# use_negative_prompt,
|
1481 |
-
# seed,
|
1482 |
-
# width,
|
1483 |
-
# height,
|
1484 |
-
# guidance_scale,
|
1485 |
-
# num_inference_steps,
|
1486 |
-
# randomize_seed,
|
1487 |
-
# num_images
|
1488 |
-
# ],
|
1489 |
-
# outputs=[result, seed],
|
1490 |
-
# api_name="run",
|
1491 |
-
# )
|
1492 |
-
run_button.click(
|
1493 |
-
fn=generate,
|
1494 |
-
inputs=[
|
1495 |
-
prompt,
|
1496 |
-
negative_prompt,
|
1497 |
-
use_negative_prompt,
|
1498 |
-
seed,
|
1499 |
-
width,
|
1500 |
-
height,
|
1501 |
-
guidance_scale,
|
1502 |
-
num_inference_steps,
|
1503 |
-
randomize_seed,
|
1504 |
-
num_images
|
1505 |
-
],
|
1506 |
-
outputs=[result, seed]
|
1507 |
-
)
|
1508 |
|
1509 |
-
###############################################################################
|
1510 |
-
############# above part is for text to image #############
|
1511 |
-
###############################################################################
|
1512 |
def clear_tts_fields():
|
1513 |
return [gr.update(value=""), gr.update(value=""), None, None, gr.update(value=False), gr.update(value=True), None, None]
|
1514 |
|
|
|
1270 |
with gr.Row():
|
1271 |
submit_tts = gr.Button(value="Submit", interactive=True)
|
1272 |
clear_tts = gr.Button(value="Clear", interactive=True)
|
1273 |
+
###############################################################################
|
1274 |
+
############# this part is for text to image #############
|
1275 |
+
###############################################################################
|
1276 |
+
|
1277 |
+
with gr.Row(variant="panel") as text2image_model:
|
1278 |
+
|
1279 |
+
with gr.Column():
|
1280 |
+
|
1281 |
+
with gr.Row():
|
1282 |
+
prompt = gr.Text(
|
1283 |
+
label="Prompt",
|
1284 |
+
show_label=False,
|
1285 |
+
max_lines=1,
|
1286 |
+
placeholder="Enter your prompt",
|
1287 |
+
container=False,
|
1288 |
+
)
|
1289 |
+
run_button = gr.Button("Run", scale=0)
|
1290 |
+
|
1291 |
+
with gr.Accordion("Advanced options", open=False):
|
1292 |
+
num_images = gr.Slider(
|
1293 |
+
label="Number of Images",
|
1294 |
+
minimum=1,
|
1295 |
+
maximum=4,
|
1296 |
+
step=1,
|
1297 |
+
value=1,
|
1298 |
+
)
|
1299 |
+
with gr.Row():
|
1300 |
+
use_negative_prompt = gr.Checkbox(label="Use negative prompt", value=True)
|
1301 |
+
negative_prompt = gr.Text(
|
1302 |
+
label="Negative prompt",
|
1303 |
+
max_lines=5,
|
1304 |
+
lines=4,
|
1305 |
+
placeholder="Enter a negative prompt",
|
1306 |
+
value="(deformed, distorted, disfigured:1.3), poorly drawn, bad anatomy, wrong anatomy, extra limb, missing limb, floating limbs, (mutated hands and fingers:1.4), disconnected limbs, mutation, mutated, ugly, disgusting, blurry, amputation, NSFW",
|
1307 |
+
visible=True,
|
1308 |
+
)
|
1309 |
+
seed = gr.Slider(
|
1310 |
+
label="Seed",
|
1311 |
+
minimum=0,
|
1312 |
+
maximum=MAX_SEED,
|
1313 |
+
step=1,
|
1314 |
+
value=0,
|
1315 |
+
)
|
1316 |
+
randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
|
1317 |
+
with gr.Row(visible=True):
|
1318 |
+
width = gr.Slider(
|
1319 |
+
label="Width",
|
1320 |
+
minimum=512,
|
1321 |
+
maximum=MAX_IMAGE_SIZE,
|
1322 |
+
step=64,
|
1323 |
+
value=1024,
|
1324 |
+
)
|
1325 |
+
height = gr.Slider(
|
1326 |
+
label="Height",
|
1327 |
+
minimum=512,
|
1328 |
+
maximum=MAX_IMAGE_SIZE,
|
1329 |
+
step=64,
|
1330 |
+
value=1024,
|
1331 |
+
)
|
1332 |
+
with gr.Row():
|
1333 |
+
guidance_scale = gr.Slider(
|
1334 |
+
label="Guidance Scale",
|
1335 |
+
minimum=0.1,
|
1336 |
+
maximum=6,
|
1337 |
+
step=0.1,
|
1338 |
+
value=3.0,
|
1339 |
+
)
|
1340 |
+
num_inference_steps = gr.Slider(
|
1341 |
+
label="Number of inference steps",
|
1342 |
+
minimum=1,
|
1343 |
+
maximum=15,
|
1344 |
+
step=1,
|
1345 |
+
value=8,
|
1346 |
+
)
|
1347 |
+
with gr.Column():
|
1348 |
+
result = gr.Gallery(label="Result", columns=2, rows=2, show_label=False)
|
1349 |
|
1350 |
|
1351 |
+
# gr.Examples(
|
1352 |
+
# examples=examples,
|
1353 |
+
# inputs=prompt,
|
1354 |
+
# cache_examples=False
|
1355 |
+
# )
|
1356 |
+
|
1357 |
+
use_negative_prompt.change(
|
1358 |
+
fn=lambda x: gr.update(visible=x),
|
1359 |
+
inputs=use_negative_prompt,
|
1360 |
+
outputs=negative_prompt,
|
1361 |
+
api_name=False,
|
1362 |
+
)
|
1363 |
+
|
1364 |
+
# gr.on(
|
1365 |
+
# triggers=[
|
1366 |
+
# prompt.submit,
|
1367 |
+
# negative_prompt.submit,
|
1368 |
+
# run_button.click,
|
1369 |
+
# ],
|
1370 |
+
# fn=generate,
|
1371 |
+
# inputs=[
|
1372 |
+
# prompt,
|
1373 |
+
# negative_prompt,
|
1374 |
+
# use_negative_prompt,
|
1375 |
+
# seed,
|
1376 |
+
# width,
|
1377 |
+
# height,
|
1378 |
+
# guidance_scale,
|
1379 |
+
# num_inference_steps,
|
1380 |
+
# randomize_seed,
|
1381 |
+
# num_images
|
1382 |
+
# ],
|
1383 |
+
# outputs=[result, seed],
|
1384 |
+
# api_name="run",
|
1385 |
+
# )
|
1386 |
+
run_button.click(
|
1387 |
+
fn=generate,
|
1388 |
+
inputs=[
|
1389 |
+
prompt,
|
1390 |
+
negative_prompt,
|
1391 |
+
use_negative_prompt,
|
1392 |
+
seed,
|
1393 |
+
width,
|
1394 |
+
height,
|
1395 |
+
guidance_scale,
|
1396 |
+
num_inference_steps,
|
1397 |
+
randomize_seed,
|
1398 |
+
num_images
|
1399 |
+
],
|
1400 |
+
outputs=[result, seed]
|
1401 |
+
)
|
1402 |
+
|
1403 |
+
###############################################################################
|
1404 |
+
############# above part is for text to image #############
|
1405 |
+
###############################################################################
|
1406 |
+
|
1407 |
|
1408 |
###############################################################################
|
1409 |
# this part is for 3d generate.
|
|
|
1514 |
# above part is for 3d generate.
|
1515 |
###############################################################################
|
1516 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1517 |
|
|
|
|
|
|
|
1518 |
def clear_tts_fields():
|
1519 |
return [gr.update(value=""), gr.update(value=""), None, None, gr.update(value=False), gr.update(value=True), None, None]
|
1520 |
|