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