fix: add types

load_models.py

@@ -1,7 +1,6 @@
 import torch
 
 import logging
-from typing import Any, Dict, List
 
 from auto_gptq import AutoGPTQForCausalLM
 from huggingface_hub import hf_hub_download

main.py

@@ -2,7 +2,7 @@ import os
 import glob
 import shutil
 import subprocess
-import asyncio
+from typing import Any, Dict, List
 
 from fastapi import FastAPI, HTTPException, UploadFile, WebSocket
 from fastapi.staticfiles import StaticFiles