Update app.py

app.py

@@ -78,7 +78,6 @@ def index(file):
         shuffle=False,
         collate_fn=lambda x: process_images(processor, x),
     )
-    ds = ["test", "double test"]
     for batch_doc in tqdm(dataloader):
         with torch.no_grad():
             batch_doc = {k: v.to(device) for k, v in batch_doc.items()}