skatzR
/

RQA-R1

Logical-Analysis

Text-Classification

Model card Files Files and versions

skatzR commited on 25 days ago

Commit

6e5c5f6

·

verified ·

1 Parent(s): 295799d

Update inference.py

Files changed (1) hide show

inference.py +1 -1

inference.py CHANGED Viewed

@@ -48,7 +48,7 @@ ERROR_THRESHOLDS = {
 # ============================================================
 class RQAJudge:
-    def __init__(self, model_name="skatzR/RQA-X1.1", device=None):
         self.device = device or ("cuda" if torch.cuda.is_available() else "cpu")
         self.tokenizer = AutoTokenizer.from_pretrained(

 # ============================================================
 class RQAJudge:
+    def __init__(self, model_name="skatzR/RQA-R1", device=None):
         self.device = device or ("cuda" if torch.cuda.is_available() else "cpu")
         self.tokenizer = AutoTokenizer.from_pretrained(