bugfix
Browse files
app.py
CHANGED
|
@@ -47,6 +47,7 @@ def process_window(frames_window):
|
|
| 47 |
|
| 48 |
inputs = processor(images=frames_window, return_tensors="pt", padding=True)
|
| 49 |
if torch.cuda.is_available():
|
|
|
|
| 50 |
model = model.to(torch.device("cuda"))
|
| 51 |
inputs = {k: v.cuda() for k, v in inputs.items()}
|
| 52 |
with torch.no_grad():
|
|
|
|
| 47 |
|
| 48 |
inputs = processor(images=frames_window, return_tensors="pt", padding=True)
|
| 49 |
if torch.cuda.is_available():
|
| 50 |
+
text_inputs = text_inputs.to(torch.device("cuda"))
|
| 51 |
model = model.to(torch.device("cuda"))
|
| 52 |
inputs = {k: v.cuda() for k, v in inputs.items()}
|
| 53 |
with torch.no_grad():
|