Update app.py

app.py

@@ -56,6 +56,8 @@ def index(files, ds):
     )
 
     print(f"model device: {model.device}")
+
+    model = model.to(model.device)
           
     for batch_doc in tqdm(dataloader):
         with torch.no_grad():