zerostratos commited on
Commit
f1a52ff
·
verified ·
1 Parent(s): 33e76db

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -9,7 +9,7 @@ import torch.nn.functional as F
9
  import pandas as pd
10
  import re
11
  from sacremoses import MosesTokenizer, MosesDetokenizer
12
- teencode_df = pd.read_csv('toxic_classification_model/teencode.txt',names=['teencode','map'],sep='\t',)
13
  teencode_list = teencode_df['teencode'].to_list()
14
  map_list = teencode_df['map'].to_list()
15
  class BCNN(nn.Module):
@@ -231,7 +231,7 @@ class TextClassificationApp:
231
 
232
  def main():
233
  # Replace these with your actual model path and class names
234
- MODEL_PATH = 'toxic_classification_model/toxic.pt'
235
  CLASS_NAMES = [
236
  'Non-toxic',
237
  'Toxic'
 
9
  import pandas as pd
10
  import re
11
  from sacremoses import MosesTokenizer, MosesDetokenizer
12
+ teencode_df = pd.read_csv('teencode.txt',names=['teencode','map'],sep='\t',)
13
  teencode_list = teencode_df['teencode'].to_list()
14
  map_list = teencode_df['map'].to_list()
15
  class BCNN(nn.Module):
 
231
 
232
  def main():
233
  # Replace these with your actual model path and class names
234
+ MODEL_PATH = 'toxic.pt'
235
  CLASS_NAMES = [
236
  'Non-toxic',
237
  'Toxic'