Hugging Face's logo

Spaces:
CUHKWilliam
/
Multiple-shots
Sleeping

App Files Community
CUHKWilliam commited on
Commit
e40c581
·
verified ·
1 Parent(s): 753655d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -11,6 +11,7 @@ import gradio as gr
11
  import time
12
  from torchvision import transforms
13
  from PIL import Image
 
14
 
15
  img_mean = [0.485, 0.456, 0.406]
16
  img_std = [0.229, 0.224, 0.225]
 
11
  import time
12
  from torchvision import transforms
13
  from PIL import Image
14
+ import torch.nn.functional as F
15
 
16
  img_mean = [0.485, 0.456, 0.406]
17
  img_std = [0.229, 0.224, 0.225]