diff --git a/app/server.py b/app/server.py index fc713eaba..864b2cb0a 100644 --- a/app/server.py +++ b/app/server.py @@ -9,10 +9,10 @@ from starlette.responses import HTMLResponse, JSONResponse from starlette.staticfiles import StaticFiles -export_file_url = 'https://www.dropbox.com/s/6bgq8t6yextloqp/export.pkl?raw=1' -export_file_name = 'export.pkl' +export_file_url = 'https://drive.google.com/uc?export=download&id=1EH87-zs52xU2ggQTXs1xwpXCi8leP7Qn' +export_file_name = 'resnet50(50epoch).pkl' -classes = ['black', 'grizzly', 'teddys'] +classes = ['AK', 'BCC', 'BKL', 'DF', 'MEL', 'NV', 'SCC', 'VASC'] path = Path(__file__).parent app = Starlette()