tmmdev commited on
Commit
3792cff
·
verified ·
1 Parent(s): 0ceeebb

Update pattern_analyzer.py

Browse files
Files changed (1) hide show
  1. pattern_analyzer.py +2 -2
pattern_analyzer.py CHANGED
@@ -6,8 +6,8 @@ from pattern_logic import PatternLogic
6
 
7
  class PatternAnalyzer:
8
  def __init__(self):
9
- self.model = AutoModelForCausalLM.from_pretrained("tmm-dev/codellama-pattern-analysis")
10
- self.tokenizer = AutoTokenizer.from_pretrained("tmm-dev/codellama-pattern-analysis")
11
  self.basic_patterns = {
12
  'channel': {'min_points': 4, 'confidence_threshold': 0.7},
13
  'triangle': {'min_points': 3, 'confidence_threshold': 0.75},
 
6
 
7
  class PatternAnalyzer:
8
  def __init__(self):
9
+ self.model = AutoModelForCausalLM.from_pretrained("tmmdev/codellama-pattern-analysis")
10
+ self.tokenizer = AutoTokenizer.from_pretrained("tmmdev/codellama-pattern-analysis")
11
  self.basic_patterns = {
12
  'channel': {'min_points': 4, 'confidence_threshold': 0.7},
13
  'triangle': {'min_points': 3, 'confidence_threshold': 0.75},