LouisLi commited on
Commit
7f283ac
·
verified ·
1 Parent(s): 2712b13

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +132 -126
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