Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 |
-
|
|
|
|
|
|
|
|
|
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] != '.':
|