PengWeixuanSZU commited on
Commit
19f7978
·
verified ·
1 Parent(s): 6f2efc1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -24,7 +24,7 @@ import PIL
24
  import torch.nn.functional as F
25
  from torchvision import transforms
26
 
27
- import spaces
28
  from huggingface_hub import snapshot_download
29
 
30
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
@@ -122,7 +122,7 @@ def init_pipe():
122
 
123
  return pipe
124
 
125
- @spaces.GPU()#duration=120
126
  def inference(source_images,
127
  target_images,
128
  text_prompt, negative_prompt,
 
24
  import torch.nn.functional as F
25
  from torchvision import transforms
26
 
27
+ #import spaces
28
  from huggingface_hub import snapshot_download
29
 
30
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 
122
 
123
  return pipe
124
 
125
+ #@spaces.GPU()#duration=120
126
  def inference(source_images,
127
  target_images,
128
  text_prompt, negative_prompt,