| from src.gui import WebUI | |
| def main(): | |
| print("Launching demo...") | |
| # cwd = "/Users/andreped/workspace/livermask/" # local testing -> macOS | |
| cwd = "/home/user/app/" # production -> docker | |
| model_name = "model.h5" # assumed to lie in `cwd` directory | |
| class_name = "parenchyma" | |
| # initialize and run app | |
| app = WebUI(model_name=model_name, class_name=class_name, cwd=cwd) | |
| app.run() | |
| if __name__ == "__main__": | |
| main() | |