Hugging Face's logo

Spaces:
RedmarkerAI
/
test_hawkeye_multi
Runtime error

App Files Community
thomascuddihy commited on
Commit
b7206e6
·
1 Parent(s): 2aa9837

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ logger = logging.getLogger(__name__)
10
  MODEL_PATH = "RedmarkerAI/hrw_multi_generous_v1"
11
  auth_token = os.environ.get("TOKEN_MODEL")
12
  dataset_token = os.environ.get("TOKEN_DATASET")
13
- hf_writer = gr.HuggingFaceDatasetSaver(dataset_token, "hrw_test_multiclass_flagged_data")
14
  model = DistilBertForSequenceClassification.from_pretrained(MODEL_PATH, use_auth_token=auth_token)
15
  tokenizer = DistilBertTokenizer.from_pretrained("distilbert-base-uncased")
16
 
 
10
  MODEL_PATH = "RedmarkerAI/hrw_multi_generous_v1"
11
  auth_token = os.environ.get("TOKEN_MODEL")
12
  dataset_token = os.environ.get("TOKEN_DATASET")
13
+ hf_writer = gr.HuggingFaceDatasetSaver(dataset_token, "RedmarkerAI/hrw_test_multiclass_flagged_data")
14
  model = DistilBertForSequenceClassification.from_pretrained(MODEL_PATH, use_auth_token=auth_token)
15
  tokenizer = DistilBertTokenizer.from_pretrained("distilbert-base-uncased")
16