Fixing the URL for the web interface

7bdeb1e7 · rprenger · 69757f9a · 7bdeb1e7
Commit 7bdeb1e7 authored Sep 20, 2021 by rprenger
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

megatron/text_generation_server.py megatron/text_generation_server.py +1 -1

No files found.
--- a/megatron/text_generation_server.py
+++ b/megatron/text_generation_server.py
@@ -82,7 +82,7 @@ class MegatronGenerate(Resource):

 class MegatronServer(object):
    def __init__(self, model):
-        self.app = Flask(__name__)
+        self.app = Flask(__name__, static_url_path='')
        api = Api(self.app)
        api.add_resource(MegatronGenerate, '/generate', resource_class_args=[model])