Spaces:

jeydipak
/

nlp_project

Runtime error

jeydipak commited on Mar 8, 2024

Commit

6369ed4

verified ·

1 Parent(s): 483fa91

Update pages/toxicapp.py

Files changed (1) hide show

pages/toxicapp.py CHANGED Viewed

@@ -17,7 +17,7 @@ model = AutoModelForSequenceClassification.from_pretrained('cointegrated/rubert-
 model.load_state_dict(torch.load(model_path, map_location=torch.device('cpu')), strict=False)
 image = Image.open("media/oritoxic.jpg")
-# df = pd.read_csv("media/Toxic_labeled.csv")
 loss_values = [0.4063596375772262, 0.402279906166038, 0.3998144585561736, 0.39567733055365567,
                0.3921396666608141, 0.38956182373070186, 0.3866641920902114, 0.3879134839351564,
                0.38288725781591604, 0.38198364493999004]

 model.load_state_dict(torch.load(model_path, map_location=torch.device('cpu')), strict=False)
 image = Image.open("media/oritoxic.jpg")
+df = pd.read_csv("media/Toxic_labeled.csv")
 loss_values = [0.4063596375772262, 0.402279906166038, 0.3998144585561736, 0.39567733055365567,
                0.3921396666608141, 0.38956182373070186, 0.3866641920902114, 0.3879134839351564,
                0.38288725781591604, 0.38198364493999004]