diegokauer commited on
Commit
40c6398
·
1 Parent(s): 8e4bead

Update _wsgi.py

Browse files
Files changed (1) hide show
  1. _wsgi.py +5 -5
_wsgi.py CHANGED
@@ -29,7 +29,7 @@ logging.config.dictConfig({
29
  })
30
 
31
  from label_studio_ml.api import init_app
32
- from model import SamMLBackend
33
 
34
  _DEFAULT_CONFIG_PATH = os.path.join(os.path.dirname(__file__), 'config.json')
35
 
@@ -101,13 +101,13 @@ if __name__ == "__main__":
101
  kwargs.update(parse_kwargs())
102
 
103
  if args.check:
104
- print('Check "' + SamMLBackend.__name__ + '" instance creation..')
105
- model = SamMLBackend(**kwargs)
106
 
107
- app = init_app(model_class=SamMLBackend)
108
 
109
  app.run(host=args.host, port=args.port, debug=args.debug)
110
 
111
  else:
112
  # for uWSGI use
113
- app = init_app(model_class=SamMLBackend)
 
29
  })
30
 
31
  from label_studio_ml.api import init_app
32
+ from model import Model
33
 
34
  _DEFAULT_CONFIG_PATH = os.path.join(os.path.dirname(__file__), 'config.json')
35
 
 
101
  kwargs.update(parse_kwargs())
102
 
103
  if args.check:
104
+ print('Check "' + Model.__name__ + '" instance creation..')
105
+ model = Model(**kwargs)
106
 
107
+ app = init_app(model_class=Model)
108
 
109
  app.run(host=args.host, port=args.port, debug=args.debug)
110
 
111
  else:
112
  # for uWSGI use
113
+ app = init_app(model_class=Model)