shainaraza
commited on
Commit
•
6be538d
1
Parent(s):
4bb8289
Update app.py
Browse files
app.py
CHANGED
@@ -37,8 +37,8 @@ class BiasPipeline:
|
|
37 |
self.classifier_model = model_loader.load_model("classification", "newsmediabias/UnBIAS-classification-bert")
|
38 |
self.ner_tokenizer = model_loader.load_tokenizer("newsmediabias/UnBIAS-NER")
|
39 |
self.ner_model = model_loader.load_model("token_classification", "newsmediabias/UnBIAS-NER")
|
40 |
-
self.classifier = tf_pipeline("text-classification", model=self.classifier_model, tokenizer=self.classifier_tokenizer, device=
|
41 |
-
self.ner = tf_pipeline("ner", model=self.ner_model, tokenizer=self.ner_tokenizer, device=
|
42 |
|
43 |
def clean_text(self, text):
|
44 |
"""Clean up the text by removing any redundant spaces."""
|
|
|
37 |
self.classifier_model = model_loader.load_model("classification", "newsmediabias/UnBIAS-classification-bert")
|
38 |
self.ner_tokenizer = model_loader.load_tokenizer("newsmediabias/UnBIAS-NER")
|
39 |
self.ner_model = model_loader.load_model("token_classification", "newsmediabias/UnBIAS-NER")
|
40 |
+
self.classifier = tf_pipeline("text-classification", model=self.classifier_model, tokenizer=self.classifier_tokenizer, device=torch_device)
|
41 |
+
self.ner = tf_pipeline("ner", model=self.ner_model, tokenizer=self.ner_tokenizer, device=torch_device)
|
42 |
|
43 |
def clean_text(self, text):
|
44 |
"""Clean up the text by removing any redundant spaces."""
|