JKJanosko commited on
Commit
fba9b41
1 Parent(s): 49a734a

fixed directories in app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -126,7 +126,7 @@ def main():
126
  # -- Creates Variables for Use of Model --
127
  attributes=["toxic","severe_toxic","obscene","threat","insult","identity_hate"]
128
  tokenizer=AutoTokenizer.from_pretrained("roberta-base")
129
- toxic_comments_dataset=toxicity_dataset("data/train.csv",tokenizer,attributes)
130
 
131
  toxicity_data_module=Toxcity_Data_Module("AppDirectory/data/train.csv","AppDirectory/data/test.csv",attributes)
132
  toxicity_data_module.setup()
 
126
  # -- Creates Variables for Use of Model --
127
  attributes=["toxic","severe_toxic","obscene","threat","insult","identity_hate"]
128
  tokenizer=AutoTokenizer.from_pretrained("roberta-base")
129
+ toxic_comments_dataset=toxicity_dataset("AppDirectory/data/train.csv",tokenizer,attributes)
130
 
131
  toxicity_data_module=Toxcity_Data_Module("AppDirectory/data/train.csv","AppDirectory/data/test.csv",attributes)
132
  toxicity_data_module.setup()