diff --git a/server_fastapi.py b/server_fastapi.py
index b7c1a52..e858f90 100644
--- a/server_fastapi.py
+++ b/server_fastapi.py
@@ -5,7 +5,6 @@ import logging
 import gc
 import random
 
-from pydantic import BaseModel
 import gradio
 import numpy as np
 import utils