Update src/predict.py

src/predict.py

@@ -27,7 +27,7 @@ class PredictionServices:
         return inputs
     
     def plot(self, pred):
-        probs = [round(pred*100,2), round(1-pred*100,2)]
+        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, 
@@ -42,7 +42,7 @@ class PredictionServices:
         fig.update_layout(yaxis_title=None,xaxis_title=None)
         st.plotly_chart(fig, theme="streamlit", use_container_width=True)
     
-    def data_validation(data):
+    def data_validation(self, data):
         df = pd.read_csv(data)
         status=True
         for column in df.columns: