Update README.md

README.md

@@ -27,18 +27,19 @@ import fasttext
 model = fasttext.load_model("model_textbook_quality.bin")
 
 
-def replace_newlines(text):
+def replace_newlines(text: str) -> str:
   return re.sub("\n+", " ", text)
 
 
-def predict(text_list: List[str]):
+def predict(text_list: List[str]) -> List[dict]:
   text_list = [replace_newlines(text) for text in text_list]
   pred = model.predict(text_list)
-  return [{"label": l[0].lstrip("__label__"), "score": s[0]} for l, s in zip(*pred)]
+  return [{"label": l[0].lstrip("__label__"), "score": s[0]}
+           for l, s in zip(*pred)]
 
 
 predict(["Hi"])
-# Output: {'label': 'LOW_QUALITY', 'score': 1.00001}
+# Output: [{'label': 'LOW_QUALITY', 'score': 1.00001}]
 
 ```