cocktailpeanut commited on
Commit
d15bc68
β€’
1 Parent(s): 4c8dced
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -15,7 +15,7 @@ from typing import List
15
  import torch
16
  import os
17
  from transformers import AutoTokenizer
18
- import spaces
19
  import numpy as np
20
  from utils_mask import get_mask_location
21
  from torchvision import transforms
@@ -123,7 +123,7 @@ pipe = TryonPipeline.from_pretrained(
123
  )
124
  pipe.unet_encoder = UNet_Encoder
125
 
126
- @spaces.GPU
127
  def start_tryon(dict,garm_img,garment_des,is_checked,is_checked_crop,denoise_steps,seed):
128
  #device = "cuda"
129
  device = devicetorch.get(torch)
 
15
  import torch
16
  import os
17
  from transformers import AutoTokenizer
18
+ #import spaces
19
  import numpy as np
20
  from utils_mask import get_mask_location
21
  from torchvision import transforms
 
123
  )
124
  pipe.unet_encoder = UNet_Encoder
125
 
126
+ #@spaces.GPU
127
  def start_tryon(dict,garm_img,garment_des,is_checked,is_checked_crop,denoise_steps,seed):
128
  #device = "cuda"
129
  device = devicetorch.get(torch)