Skip to content

Commit

Permalink
add rcfile
Browse files Browse the repository at this point in the history
  • Loading branch information
nlueem committed Aug 3, 2024
1 parent 17c1d35 commit 1ae8493
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 2 deletions.
2 changes: 2 additions & 0 deletions .pylintrc
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
[MASTER]
disable=import-error
4 changes: 2 additions & 2 deletions app.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
"""
This module sets up a FastAPI server to interact with a text-generation model.
It uses Hugging Face transformers, Pydantic for request validation, and Torch for device management.
It uses Hugging Face transformers, pydantic for request validation, and Torch for device management.
"""

import os
Expand Down Expand Up @@ -73,7 +73,7 @@

# Data model for making POST requests to /chat
class ChatRequest(BaseModel):
"""Class representing a data-model"""
"""Class representing the data-model"""
messages: list
temperature: Union[float, None] = None
top_p: Union[float, None] = None
Expand Down

0 comments on commit 1ae8493

Please sign in to comment.