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

Fixing the URL for the web interface

parent 69757f9a

megatron/text_generation_server.py

+1 −1
Original line number Diff line number Diff line
@@ -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])