diff --git a/api/DataProcesser.py b/api/DataProcesser.py
index 3934be55..2ba3f577 100644
--- a/api/DataProcesser.py
+++ b/api/DataProcesser.py
@@ -84,6 +84,7 @@ def pretrained_predict(self, df, model_name):
             pipeline = pickle.load(model)
 
         texts_to_predict = df['input_column']
+        texts_to_predict = [str(text) for text in texts_to_predict]
         predictions = pipeline.predict(texts_to_predict)
         df['output_column'] = predictions
         return df
diff --git a/api/Neural_Network2.py b/api/Neural_Network2.py
index 4cb9f08b..ba1b3106 100644
--- a/api/Neural_Network2.py
+++ b/api/Neural_Network2.py
@@ -34,9 +34,10 @@ def __init__(self, epochs):
         self.epochs = epochs
         self.epoch_num = 0
 
-
     def on_epoch_begin(self, epoch, logs=None):
         self.epoch_step = 0
+
+    def on_epoch_end(self, epoch, logs=None):
         self.epoch_num += 1
 
     def on_batch_end(self, batch, logs=None):
diff --git a/api/app.py b/api/app.py
index 625e5440..c8489eaa 100644
--- a/api/app.py
+++ b/api/app.py
@@ -6,13 +6,10 @@
 
 import time
 import os
-import atexit
-import threading
 import pandas as pd
 import nltk
 import json
 import asyncio
-import logging
 nltk.download('wordnet')
 
 
diff --git a/requirements.txt b/requirements.txt
index 373f6b42..6eb08c56 100644
Binary files a/requirements.txt and b/requirements.txt differ