Update epi_pipeline.py

epi_pipeline.py

@@ -25,7 +25,7 @@ def timeit(func):
         end_time = time.perf_counter()
         total_time = end_time - start_time
         # first item in the args, ie `args[0]` is `self`
-        print(f'Function {func.__name__}{args} took {total_time:.4f} seconds to execute')
+        print(f'Function {func.__name__} took {total_time:.4f} seconds to execute')
         return result
     return timeit_wrapper
 
@@ -273,7 +273,6 @@ 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