Shoubin commited on
Commit
170b316
1 Parent(s): 98739d8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -84,7 +84,11 @@ def sevila_demo(video,
84
  clip_proposal=None
85
  )
86
  clip = transform(raw_clip.permute(1,0,2,3))
87
- clip = clip.float().to(int(device))
 
 
 
 
88
  clip = clip.unsqueeze(0)
89
  # check
90
  if option1[-1] != '.':
 
84
  clip_proposal=None
85
  )
86
  clip = transform(raw_clip.permute(1,0,2,3))
87
+ if device == "cpu":
88
+ clip = clip.float()
89
+ else:
90
+ clip = clip.float().to(int(device))
91
+
92
  clip = clip.unsqueeze(0)
93
  # check
94
  if option1[-1] != '.':