diff --git a/examples/dstc9/dstc9_server.py b/examples/dstc9/dstc9_server.py index 1977ebf2..aea64a24 100644 --- a/examples/dstc9/dstc9_server.py +++ b/examples/dstc9/dstc9_server.py @@ -51,7 +51,7 @@ def generate_for_queue(in_queue, out_queue): if __name__ == "__main__": - from DialoGLM.server import * + from GODEL.server import * # replace the path with your trained checkpoint args.model_name_or_path = 't5-base' main() diff --git a/requirements.txt b/requirements.txt index e85cc277..757e1157 100644 --- a/requirements.txt +++ b/requirements.txt @@ -16,7 +16,6 @@ fire==0.4.0 flask==2.0.2 flask-Cors==3.0.10 chardet==3.0.4 -nltk==3.4.1 beautifulsoup4==4.6.0 zstandard==0.18.0 flashtext==2.7