Update app.py

app.py

@@ -48,12 +48,12 @@ def check_prime_number(num: int) -> str:
             # If num is divisible by any number between
             # 2 and n / 2, it is not prime
             if (num % i) == 0:
-                print(num, "is not a prime number")
+                return f"{num} is not a prime number"
                 break
             else:
-                print(num, "is a prime number")
+                return f"{num} is a prime number"
     else:
-        print(num, "is not a prime number")
+        return f"{num} is not a prime number"
 
 
 final_answer = FinalAnswerTool()