Update app.py

app.py

@@ -14,7 +14,7 @@ class ONNXInferencePipeline:
             raise ValueError("HF_TOKEN environment variable is not set.")
 
         # Download files from Hugging Face Hub using the token
-        self.onnx_path = hf_hub_download(repo_id=repo_id, filename="RudeRater.onnx", use_auth_token=hf_token)
+        self.onnx_path = hf_hub_download(repo_id=repo_id, filename="model.onnx", use_auth_token=hf_token)
         self.tokenizer_path = hf_hub_download(repo_id=repo_id, filename="train_bpe_tokenizer.json", use_auth_token=hf_token)
         self.config_path = hf_hub_download(repo_id=repo_id, filename="hyperparameters.json", use_auth_token=hf_token)
 
@@ -65,7 +65,7 @@ class ONNXInferencePipeline:
 # Example usage
 if __name__ == "__main__":
     # Initialize the pipeline with the Hugging Face repository ID
-    pipeline = ONNXInferencePipeline(repo_id="iimran/RudeRater")
+    pipeline = ONNXInferencePipeline(repo_id="iimran/abuse-detector")
 
     # Example texts for testing
     example_texts = [