Update src/predict.py

src/predict.py

@@ -32,7 +32,7 @@ class PredictionServices:
         color_map = {'toxic':"red", "non-toxic":"green"}
         fig = px.bar(x=probs, 
                      y=labels,
-                     width=600, height=250, 
+                     width=400, height=250, 
                      template="plotly_dark", 
                      text_auto=True, 
                      title="Probabilities(%)",
@@ -63,7 +63,7 @@ class PredictionServices:
                     df['probabilities'] = preds
                     df['toxic'] = np.where(df['probabilities']>0.5, 1, 0)
                 st.success("Prediction Process Completed!!!, :thumbsup:")
-                st.info("Prediction file will be downloaded")
+                st.info("Press download button to download prediction file")
                 return df
             else:
                 st.error("Data Validation Failed!! :thumbsdown:")
@@ -73,9 +73,9 @@ class PredictionServices:
     def single_predict(self, text:str):
         try:
             text = re.sub('\n',' ',text).strip().lower()
-            with st.spinner('Please Wait!!! Prediction in process....'):
-                input = self.tokenizer_fn(text)
-                pred = self.model.predict(input)[0][0]
+            
+            input = self.tokenizer_fn(text)
+            pred = self.model.predict(input)[0][0]
             return pred
         except Exception as e:
             print(e)