Update app.py

app.py

@@ -37,8 +37,8 @@ class BiasPipeline:
         self.classifier_model = model_loader.load_model("classification", "newsmediabias/UnBIAS-classification-bert")
         self.ner_tokenizer = model_loader.load_tokenizer("newsmediabias/UnBIAS-NER")
         self.ner_model = model_loader.load_model("token_classification", "newsmediabias/UnBIAS-NER")
-        self.classifier = tf_pipeline("text-classification", model=self.classifier_model, tokenizer=self.classifier_tokenizer, device=0)
-        self.ner = tf_pipeline("ner", model=self.ner_model, tokenizer=self.ner_tokenizer, device=0)
+        self.classifier = tf_pipeline("text-classification", model=self.classifier_model, tokenizer=self.classifier_tokenizer, device=torch_device)
+        self.ner = tf_pipeline("ner", model=self.ner_model, tokenizer=self.ner_tokenizer, device=torch_device)
 
     def clean_text(self, text):
         """Clean up the text by removing any redundant spaces."""