Gainward777 commited on
Commit
4022ec1
·
verified ·
1 Parent(s): bc46bfc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -14,6 +14,7 @@ from transformers import AutoModelForMaskGeneration, AutoProcessor, pipeline
14
  from PIL import Image
15
  import os
16
  import gc
 
17
 
18
 
19
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
@@ -111,4 +112,7 @@ gr.Interface(
111
  outputs = [
112
  gr.Image()
113
  ]
 
 
 
114
  ).launch(debug=True,share=True)
 
14
  from PIL import Image
15
  import os
16
  import gc
17
+ import glob
18
 
19
 
20
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 
112
  outputs = [
113
  gr.Image()
114
  ]
115
+ examples=[
116
+ ["./examples/models/girl1.jpg", "./examples/clothes/t_short.jpg", "shirt", "shirt"],
117
+ ],
118
  ).launch(debug=True,share=True)