diff --git a/bot.py b/bot.py
index 2bfcb2b..2f0ef5d 100755
--- a/bot.py
+++ b/bot.py
@@ -129,9 +129,9 @@ def text_handler(bot, update):
             'row1': [data['month'], data['day'], data['temp'], data['RH'], data['speed'], data['rain']]
         }
         sample = pd.DataFrame.from_dict(sample, orient='index')
-        print(sample)
-        # prediction = model.predict(sample)
-        prediction = [4.0]
+        # print(sample)
+        prediction = model.predict(sample)
+        # prediction = [4.0]
 
         data['firearea'] = round(prediction[0])
         data['nearest'] = round(min_distance(float(location['latitude']), float(location['longitude']))[0])
diff --git a/requirements.txt b/requirements.txt
index 9c6524c..90d63ac 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,5 +1,5 @@
 asn1crypto==0.24.0
-catboost==0.10.3
+catboost==0.4.1
 certifi==2018.10.15
 cffi==1.11.5
 chardet==3.0.4