Update app.py

app.py

@@ -34,9 +34,6 @@ def search(query: str, ds, images, k):
     if device != model.device:
         model.to(device)
         
-    print(f"model device: {model.device}")
-
-
     qs = []
     with torch.no_grad():
         batch_query = process_queries(processor, [query], mock_image)
@@ -79,14 +76,10 @@ def index(files, ds):
     if device != model.device:
         model.to(device)
         
-    print(f"model device: {model.device}")
-
           
     for batch_doc in tqdm(dataloader):
         with torch.no_grad():
             batch_doc = {k: v.to(device) for k, v in batch_doc.items()}
-            print(f"model device: {model.device}")
-            print(f"model device: {batch_doc['input_ids']}")
             embeddings_doc = model(**batch_doc)
         ds.extend(list(torch.unbind(embeddings_doc.to("cpu"))))
     return f"Uploaded and converted {len(images)} pages", ds, images