Update main.py

main.py

@@ -16,9 +16,9 @@ class Validation(BaseModel):
     user_prompt: str  # User's input prompt
     system_prompt: str  # System's guiding prompt
     max_tokens: int = 1024,
-    temperature: int = 0.001,
-    top_p: int = 0.9,
-    repeat_penalty: int = 1.1,
+    temperature: float = 0.001,
+    top_p: float = 0.9,
+    repeat_penalty: float = 1.1,
     top_k: int = 40
 
 # FastAPI application initialization