Refactored example code a bit

README.md

@@ -62,7 +62,9 @@ model = ClassificationModel(
     
 )
 
-predictions, logit_output = model.predict(["Build the wall", "Build the wall of trust"])
+predictions, logit_output = model.predict(["Build the wall", 
+                                        "Build the wall of trust"]
+                                        )
 predictions
 ### Output:
 ### array([1, 0])