r0903

nile4000 · Aug 3, 2024 · b94eb13 · b94eb13
1 parent 1ae8493
commit b94eb13
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/app.py b/app.py
@@ -72,12 +72,14 @@
 logger.info("Model is loaded")
 
 # Data model for making POST requests to /chat
+# pylint: disable=R0903
 class ChatRequest(BaseModel):
     """Class representing the data-model"""
     messages: list
     temperature: Union[float, None] = None
     top_p: Union[float, None] = None
     max_new_tokens: Union[int, None] = None
+# pylint: enable=R0903
 
 
 def generate(messages: list,