Spaces:

Namgyu-Youn
/

topic-generator

Sleeping

Namgyu-Youn commited on Jan 4

Commit

4de9003

verified ·

1 Parent(s): 33d44cd

Update scripts/analyzer.py

Files changed (1) hide show

scripts/analyzer.py CHANGED Viewed

@@ -5,7 +5,7 @@ from .topic_list import TOPIC_LIST
 class TopicAnalyzer:
     def __init__(self):
         self.device = "cpu"
-        self.model_name = "microsoft/deberta-v3-base"
         self.tokenizer = None
         self.model = None
         self.classifier = None

 class TopicAnalyzer:
     def __init__(self):
         self.device = "cpu"
+        self.model_name = "facebook/bart-large-mnli"
         self.tokenizer = None
         self.model = None
         self.classifier = None