Kuberwastaken commited on
Commit
a7a10f6
·
1 Parent(s): 36c279f

Minor Adjustments

Browse files
Files changed (1) hide show
  1. model/analyzer.py +2 -2
model/analyzer.py CHANGED
@@ -106,7 +106,7 @@ class ContentAnalyzer:
106
  **inputs,
107
  max_new_tokens=2,
108
  do_sample=True,
109
- temperature=0.4,
110
  top_p=0.9,
111
  pad_token_id=self.tokenizer.eos_token_id
112
  )
@@ -177,7 +177,7 @@ class ContentAnalyzer:
177
  "Vomit": {
178
  "mapped_name": "Vomit",
179
  "description": (
180
- "Any explicit reference to vomiting or related actions. This includes very specific mentions of nausea or the act of vomiting, with more focus on the direct description."
181
  )
182
  },
183
 
 
106
  **inputs,
107
  max_new_tokens=2,
108
  do_sample=True,
109
+ temperature=0.3,
110
  top_p=0.9,
111
  pad_token_id=self.tokenizer.eos_token_id
112
  )
 
177
  "Vomit": {
178
  "mapped_name": "Vomit",
179
  "description": (
180
+ "Any explicit reference to vomiting or related actions. This includes only very specific mentions of nausea or the act of vomiting, with more focus on the direct description, only flag this if you absolutely believe it's present."
181
  )
182
  },
183