ricardocavaretti commited on
Commit
142d5a5
·
verified ·
1 Parent(s): ba6ca56

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -64,6 +64,7 @@ tokenizer_two = AutoTokenizer.from_pretrained(
64
  noise_scheduler = DDPMScheduler.from_pretrained(base_path, subfolder="scheduler")
65
 
66
  text_encoder_one = CLIPTextModel.from_pretrained(
 
67
  base_path,
68
  subfolder="text_encoder",
69
  torch_dtype=torch.float16,
@@ -106,6 +107,7 @@ tensor_transfrom = transforms.Compose(
106
  ]
107
  )
108
 
 
109
  pipe = TryonPipeline.from_pretrained(
110
  base_path,
111
  unet=unet,
@@ -260,8 +262,8 @@ for ex_human in human_list_path:
260
 
261
  image_blocks = gr.Blocks(theme="Nymbo/Alyx_Theme").queue()
262
  with image_blocks as demo:
263
- gr.HTML("<center><h1>Virtual Try-On</h1></center>")
264
- gr.HTML("<center><p>Upload an image of a person and an image of a garment ✨</p></center>")
265
  with gr.Row():
266
  with gr.Column():
267
  imgs = gr.ImageEditor(sources='upload', type="pil", label='Human. Mask with pen or use auto-masking', interactive=True)
 
64
  noise_scheduler = DDPMScheduler.from_pretrained(base_path, subfolder="scheduler")
65
 
66
  text_encoder_one = CLIPTextModel.from_pretrained(
67
+
68
  base_path,
69
  subfolder="text_encoder",
70
  torch_dtype=torch.float16,
 
107
  ]
108
  )
109
 
110
+
111
  pipe = TryonPipeline.from_pretrained(
112
  base_path,
113
  unet=unet,
 
262
 
263
  image_blocks = gr.Blocks(theme="Nymbo/Alyx_Theme").queue()
264
  with image_blocks as demo:
265
+ gr.HTML("<center><h1>Loja imersiva</h1></center>")
266
+ gr.HTML("<center><p>suba sua imagem e a imagem de um produto✨</p></center>")
267
  with gr.Row():
268
  with gr.Column():
269
  imgs = gr.ImageEditor(sources='upload', type="pil", label='Human. Mask with pen or use auto-masking', interactive=True)