Update app.py

app.py

@@ -88,7 +88,7 @@ def add_pdf_gradio(pdf_file_binary, progress=gr.Progress()):
         image_md5 = get_image_md5(image)
         image_md5s.append(image_md5)
         with torch.no_grad():
-            reps = encode([image])
+            reps = encode([image]).squeeze(0)
         reps_list.append(reps)
         images.append(image)