Update app.py
Browse files
app.py
CHANGED
|
@@ -56,6 +56,8 @@ def index(files, ds):
|
|
| 56 |
)
|
| 57 |
|
| 58 |
print(f"model device: {model.device}")
|
|
|
|
|
|
|
| 59 |
|
| 60 |
for batch_doc in tqdm(dataloader):
|
| 61 |
with torch.no_grad():
|
|
|
|
| 56 |
)
|
| 57 |
|
| 58 |
print(f"model device: {model.device}")
|
| 59 |
+
|
| 60 |
+
model = model.to(model.device)
|
| 61 |
|
| 62 |
for batch_doc in tqdm(dataloader):
|
| 63 |
with torch.no_grad():
|