Spaces:

shivansh-ka
/

Toxic-Comment-Classifier

Build error

shivansh-ka commited on May 14, 2023

Commit

b29791f

1 Parent(s): 4dbcfd6

Update src/predict.py

Files changed (1) hide show

src/predict.py CHANGED Viewed

@@ -29,7 +29,7 @@ class PredictionServices:
     def plot(self, pred):
         probs = [round(pred*100,2), round(1-pred*100,2)]
         labels = ['toxic', 'non-toxic']
-        color_map - {'toxic':"red", "non-toxic":"green"}
         fig = px.bar(x=probs,
                      y=labels,
                      width=500, height=250,

     def plot(self, pred):
         probs = [round(pred*100,2), round(1-pred*100,2)]
         labels = ['toxic', 'non-toxic']
+        color_map = {'toxic':"red", "non-toxic":"green"}
         fig = px.bar(x=probs,
                      y=labels,
                      width=500, height=250,