Update epi_pipeline.py

epi_pipeline.py

@@ -253,6 +253,7 @@ class Classify_Pipeline:
             prob_tensor = output.logits.softmax(dim=-1)
             # We only want to return the probability that it is true
             prob = prob_tensor.data[0][1].item()
+            print(f"Abstract is epidemiological: {isEpi}")
             return prob, isEpi
         else:
             return 0.0, False