diff --git a/web.py b/web.py
index 369b86b..7de33c8 100644
--- a/web.py
+++ b/web.py
@@ -7,7 +7,7 @@
 class Input(BaseModel):
     generate_tokens_limit: int = 100
     top_p: float = 0.7
-    top_k: float = 0
+    top_k: int = 0
     temperature: float = 1.0
     text: str