Kuberwastaken commited on
Commit
8281640
·
1 Parent(s): e503e0f

Minor Improvements

Browse files
model/__pycache__/analyzer.cpython-310.pyc CHANGED
Binary files a/model/__pycache__/analyzer.cpython-310.pyc and b/model/__pycache__/analyzer.cpython-310.pyc differ
 
model/analyzer.py CHANGED
@@ -89,7 +89,7 @@ class ContentAnalyzer:
89
  print(f"\nAnalyzing for {mapped_name}...")
90
  prompt = f"""
91
  Check this text for any clear indication of {mapped_name} ({description}).
92
- Be sensitive to subtle references or implications, make sure the text is not metaphorical.
93
  Respond concisely and only with: YES, NO, or MAYBE.
94
  Text: {chunk}
95
  Answer:
@@ -106,8 +106,8 @@ class ContentAnalyzer:
106
  **inputs,
107
  max_new_tokens=2,
108
  do_sample=True,
109
- temperature=0.4,
110
- top_p=0.8,
111
  pad_token_id=self.tokenizer.eos_token_id
112
  )
113
 
 
89
  print(f"\nAnalyzing for {mapped_name}...")
90
  prompt = f"""
91
  Check this text for any clear indication of {mapped_name} ({description}).
92
+ only say yes if you are confident, make sure the text is not metaphorical.
93
  Respond concisely and only with: YES, NO, or MAYBE.
94
  Text: {chunk}
95
  Answer:
 
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
  )
113