Merge branch 'master' of https://github.com/R1j1t/contextualSpellCheck

R1j1t · Aug 23, 2020 · ef4c13e · ef4c13e
2 parents f90d071 + d653a36
commit ef4c13e
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/RESTAPI/modelAPI.py b/RESTAPI/modelAPI.py
@@ -1,8 +1,7 @@
-from flask import request
 from contextualSpellCheck.contextualSpellCheck import ContextualSpellCheck
 import spacy
 import json
-from flask import Flask, render_template, request, make_response, jsonify
+from flask import Flask, request, make_response
 
 
 # url

diff --git a/requirements.txt b/requirements.txt
@@ -1,6 +1,6 @@
-torch
-editdistance
 Flask
-transformers
 spacy
+editdistance
 pytest
+torch
+transformers
diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 
 setuptools.setup(
     name="contextualSpellCheck",
-    version="0.2.0",
+    version="0.2.1",
     author="R1j1t",
     author_email="r1j1t@protonmail.com",
     description="Contextual spell correction using BERT (bidirectional representations)",
@@ -28,4 +28,5 @@
         "Development Status :: 4 - Beta",
     ],
     python_requires=">=3.6",
+    install_requires=["torch", "editdistance", "transformers", "spacy"],
 )