Spaces:

Kuberwastaken
/

TREAT

Running

Kuberwastaken commited on Jan 10

Commit

a4a432f

1 Parent(s): 8281640

Minor Improvements

Files changed (1) hide show

model/analyzer.py CHANGED Viewed

@@ -106,7 +106,7 @@ class ContentAnalyzer:
                         **inputs,
                         max_new_tokens=2,
                         do_sample=True,
-                        temperature=0.3,
                         top_p=0.9,
                         pad_token_id=self.tokenizer.eos_token_id
                     )

                         **inputs,
                         max_new_tokens=2,
                         do_sample=True,
+                        temperature=0.4,
                         top_p=0.9,
                         pad_token_id=self.tokenizer.eos_token_id
                     )