Update app.py

app.py

@@ -4,10 +4,10 @@ from transformers import DistilBertForSequenceClassification
 from transformers import pipeline
 import gradio as gr
 
-MODEL_PATH = "xiaopeiwu/hrw_v2"
+MODEL_PATH = "RedmarkerAI/hrw_v2"
 auth_token = os.environ.get("TOKEN_FROM_SECRET")
 dataset_token = os.environ.get("DATASET_TOKEN")
-hf_writer = gr.HuggingFaceDatasetSaver(dataset_token, "hrw_v2_test_flagged_data")
+hf_writer = gr.HuggingFaceDatasetSaver(dataset_token, "hrw_test_binary_flagged_data")
 model = DistilBertForSequenceClassification.from_pretrained(MODEL_PATH, use_auth_token=auth_token)
 tokenizer = DistilBertTokenizer.from_pretrained("distilbert-base-uncased")