From dc3bb5ee38d92422335612321efcc8656cdff35d Mon Sep 17 00:00:00 2001 From: Lucien Date: Sat, 25 Mar 2023 20:24:31 +0800 Subject: [PATCH] remove mock --- websocket_api.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/websocket_api.py b/websocket_api.py index 4aae12d..8a33fcd 100644 --- a/websocket_api.py +++ b/websocket_api.py @@ -1,6 +1,6 @@ from fastapi import FastAPI, WebSocket, WebSocketDisconnect from fastapi.responses import HTMLResponse -from mock_transformers import AutoTokenizer, AutoModel +from transformers import AutoTokenizer, AutoModel import uvicorn