Skip to content

GitLab

Commit 7bdeb1e7 authored by rprenger's avatar rprenger
Browse files

Fixing the URL for the web interface

parent 69757f9a
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
megatron/text_generation_server.py
View file @ 7bdeb1e7
...@@ -82,7 +82,7 @@ class MegatronGenerate(Resource): ...@@ -82,7 +82,7 @@ class MegatronGenerate(Resource):
class MegatronServer(object): class MegatronServer(object):
def __init__(self, model): def __init__(self, model):
self.app = Flask(__name__) self.app = Flask(__name__, static_url_path='')
api = Api(self.app) api = Api(self.app)
api.add_resource(MegatronGenerate, '/generate', resource_class_args=[model]) api.add_resource(MegatronGenerate, '/generate', resource_class_args=[model])
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment