thomascuddihy commited on
Commit
a34a62f
·
1 Parent(s): af4e4e5

fix: Update dataset path

Browse files

REquired an update after trasnfer from XP

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ import gradio as gr
7
  MODEL_PATH = "RedmarkerAI/hrw_v2"
8
  auth_token = os.environ.get("TOKEN_FROM_SECRET")
9
  dataset_token = os.environ.get("DATASET_TOKEN")
10
- hf_writer = gr.HuggingFaceDatasetSaver(dataset_token, "hrw_test_binary_flagged_data")
11
  model = DistilBertForSequenceClassification.from_pretrained(MODEL_PATH, use_auth_token=auth_token)
12
  tokenizer = DistilBertTokenizer.from_pretrained("distilbert-base-uncased")
13
 
 
7
  MODEL_PATH = "RedmarkerAI/hrw_v2"
8
  auth_token = os.environ.get("TOKEN_FROM_SECRET")
9
  dataset_token = os.environ.get("DATASET_TOKEN")
10
+ hf_writer = gr.HuggingFaceDatasetSaver(dataset_token, "RedmarkerAI/hrw_test_binary_flagged_data")
11
  model = DistilBertForSequenceClassification.from_pretrained(MODEL_PATH, use_auth_token=auth_token)
12
  tokenizer = DistilBertTokenizer.from_pretrained("distilbert-base-uncased")
13