LiangbinXie commited on
Commit
b0c2178
1 Parent(s): 4d64269

update canny.py

Browse files
Files changed (1) hide show
  1. app.py +64 -13
app.py CHANGED
@@ -36,6 +36,23 @@ urls = {
36
  'andite/anything-v4.0': ['anything-v4.0-pruned.ckpt', 'anything-v4.0.vae.pt'],
37
  }
38
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
39
  if os.path.exists('models') == False:
40
  os.mkdir('models')
41
  for repo in urls:
@@ -305,18 +322,18 @@ with gr.Blocks(css='style.css') as demo:
305
  "Nothing",
306
  "Nothing",
307
  "Nothing",
308
- "https://user-images.githubusercontent.com/52127135/223114920-cae3e723-3683-424a-bebc-0875479f2409.jpg",
309
- "https://user-images.githubusercontent.com/52127135/223121793-20c2ac6a-5a4f-4ff8-88ea-6d007a7959dd.png",
310
- "https://user-images.githubusercontent.com/52127135/223114946-6ccc127f-cb58-443e-8677-805f5dbaf6f1.png",
311
- "https://user-images.githubusercontent.com/52127135/223121793-20c2ac6a-5a4f-4ff8-88ea-6d007a7959dd.png",
312
- "https://user-images.githubusercontent.com/52127135/223121793-20c2ac6a-5a4f-4ff8-88ea-6d007a7959dd.png",
313
- "https://user-images.githubusercontent.com/52127135/223121793-20c2ac6a-5a4f-4ff8-88ea-6d007a7959dd.png",
314
- "https://user-images.githubusercontent.com/52127135/223121793-20c2ac6a-5a4f-4ff8-88ea-6d007a7959dd.png",
315
- "https://user-images.githubusercontent.com/52127135/223121793-20c2ac6a-5a4f-4ff8-88ea-6d007a7959dd.png",
316
- "https://user-images.githubusercontent.com/52127135/223121793-20c2ac6a-5a4f-4ff8-88ea-6d007a7959dd.png",
317
- "https://user-images.githubusercontent.com/52127135/223121793-20c2ac6a-5a4f-4ff8-88ea-6d007a7959dd.png",
318
- "https://user-images.githubusercontent.com/52127135/223121793-20c2ac6a-5a4f-4ff8-88ea-6d007a7959dd.png",
319
- "https://user-images.githubusercontent.com/52127135/223121793-20c2ac6a-5a4f-4ff8-88ea-6d007a7959dd.png",
320
  1,
321
  1,
322
  1,
@@ -327,7 +344,41 @@ with gr.Blocks(css='style.css') as demo:
327
  "longbody, lowres, bad anatomy, bad hands, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality",
328
  7.5,
329
  1,
330
- 3000,
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
331
  50,
332
  512,
333
  1,
 
36
  'andite/anything-v4.0': ['anything-v4.0-pruned.ckpt', 'anything-v4.0.vae.pt'],
37
  }
38
 
39
+ # download image samples
40
+ torch.hub.download_url_to_file(
41
+ 'https://user-images.githubusercontent.com/52127135/223114920-cae3e723-3683-424a-bebc-0875479f2409.jpg',
42
+ 'cyber_style.jpg')
43
+ torch.hub.download_url_to_file(
44
+ 'https://user-images.githubusercontent.com/52127135/223114946-6ccc127f-cb58-443e-8677-805f5dbaf6f1.png',
45
+ 'sword.png')
46
+ torch.hub.download_url_to_file(
47
+ 'https://user-images.githubusercontent.com/52127135/223121793-20c2ac6a-5a4f-4ff8-88ea-6d007a7959dd.png',
48
+ 'white.png')
49
+ torch.hub.download_url_to_file(
50
+ 'https://user-images.githubusercontent.com/52127135/223127404-4a3748cf-85a6-40f3-af31-a74e206db96e.jpeg',
51
+ 'scream_style.jpeg')
52
+ torch.hub.download_url_to_file(
53
+ 'https://user-images.githubusercontent.com/52127135/223127433-8768913f-9872-4d24-b883-a19a3eb20623.jpg',
54
+ 'motorcycle.jpg')
55
+
56
  if os.path.exists('models') == False:
57
  os.mkdir('models')
58
  for repo in urls:
 
322
  "Nothing",
323
  "Nothing",
324
  "Nothing",
325
+ "cyber_style.jpg",
326
+ "white.png",
327
+ "sword.png",
328
+ "white.png",
329
+ "white.png",
330
+ "white.png",
331
+ "white.png",
332
+ "white.png",
333
+ "white.png",
334
+ "white.png",
335
+ "white.png",
336
+ "white.png",
337
  1,
338
  1,
339
  1,
 
344
  "longbody, lowres, bad anatomy, bad hands, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality",
345
  7.5,
346
  1,
347
+ 2500,
348
+ 50,
349
+ 512,
350
+ 1,
351
+ ],
352
+ [
353
+ "Image",
354
+ "Nothing",
355
+ "Image",
356
+ "Nothing",
357
+ "Nothing",
358
+ "Nothing",
359
+ "scream_style.jpeg",
360
+ "white.png",
361
+ "motorcycle.jpg",
362
+ "white.png",
363
+ "white.png",
364
+ "white.png",
365
+ "white.png",
366
+ "white.png",
367
+ "white.png",
368
+ "white.png",
369
+ "white.png",
370
+ "white.png",
371
+ 1,
372
+ 1,
373
+ 1,
374
+ 1,
375
+ 1,
376
+ 1,
377
+ "motorcycle",
378
+ "longbody, lowres, bad anatomy, bad hands, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality",
379
+ 7.5,
380
+ 1,
381
+ 2500,
382
  50,
383
  512,
384
  1,