Spaces:
Runtime error
Runtime error
nanushio commited on
Commit ·
a877246
1
Parent(s): c0d774a
- [MINOR] [SOURCE] [UPDATE] 1. update app.py
Browse files
cover/models/evaluator.py
CHANGED
|
@@ -142,7 +142,7 @@ class COVER(nn.Module):
|
|
| 142 |
else:
|
| 143 |
scores += [getattr(self, "vqa_head")(feat_gated)]
|
| 144 |
elif key == 'semantic':
|
| 145 |
-
x = vclips[key]
|
| 146 |
x = x.permute(1,0,2,3)
|
| 147 |
feat, _ = getattr(self, key.split("_")[0] + "_backbone")(
|
| 148 |
x, multi=self.multi, layer=self.layer, **kwargs
|
|
|
|
| 142 |
else:
|
| 143 |
scores += [getattr(self, "vqa_head")(feat_gated)]
|
| 144 |
elif key == 'semantic':
|
| 145 |
+
x = vclips[key].squeeze(0)
|
| 146 |
x = x.permute(1,0,2,3)
|
| 147 |
feat, _ = getattr(self, key.split("_")[0] + "_backbone")(
|
| 148 |
x, multi=self.multi, layer=self.layer, **kwargs
|