Spaces:
Runtime error
Runtime error
Update pattern_analyzer.py
Browse files- 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("
|
| 10 |
-
self.tokenizer = AutoTokenizer.from_pretrained("
|
| 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},
|