Spaces:
Sleeping
Sleeping
Update main.py
Browse files
main.py
CHANGED
|
@@ -39,7 +39,7 @@ class Item(BaseModel):
|
|
| 39 |
max_new_tokens: int = 1048 # Maximum response length
|
| 40 |
top_p: float = 0.9 # Sampling parameter for diverse responses
|
| 41 |
repetition_penalty: float = 1.1 # Prevents repetition
|
| 42 |
-
key: str = "HF_API_TOKEN" # API key if needed
|
| 43 |
|
| 44 |
# Define rejection responses
|
| 45 |
rejection_responses = [
|
|
|
|
| 39 |
max_new_tokens: int = 1048 # Maximum response length
|
| 40 |
top_p: float = 0.9 # Sampling parameter for diverse responses
|
| 41 |
repetition_penalty: float = 1.1 # Prevents repetition
|
| 42 |
+
key: str = ["HF_API_TOKEN"] # API key if needed
|
| 43 |
|
| 44 |
# Define rejection responses
|
| 45 |
rejection_responses = [
|