KaushikShresth12 commited on
Commit
b85b2f9
1 Parent(s): ba6bcc6

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -3
main.py CHANGED
@@ -12,9 +12,8 @@ API_URL = "https://api-inference.huggingface.co/models/mistralai/Mixtral-8x7B-In
12
  class Item(BaseModel):
13
  prompt: str
14
  history: list
15
- system_prompt: str
16
- temperature: float = 0.0
17
- max_new_tokens: int = 1048
18
  top_p: float = 0.15
19
  repetition_penalty: float = 1.0
20
  instructions: str = ""
 
12
  class Item(BaseModel):
13
  prompt: str
14
  history: list
15
+ temperature: float = 0.1
16
+ max_new_tokens: int = 2
 
17
  top_p: float = 0.15
18
  repetition_penalty: float = 1.0
19
  instructions: str = ""