From 9be608f195b1b5aa2d0a3c560927aa9c7c213030 Mon Sep 17 00:00:00 2001
From: Cameron <cameron.maloney@warriorlife.net>
Date: Mon, 1 Apr 2024 18:45:11 -0700
Subject: [PATCH] Working base for tensorflow without prediction of trained
 models

---
 api/DataProcesser.py   |   1 +
 api/Neural_Network2.py |   3 ++-
 api/app.py             |   3 ---
 requirements.txt       | Bin 332 -> 1143 bytes
 4 files changed, 3 insertions(+), 4 deletions(-)

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 373f6b42a6a96b63535b766a06ae81c825de5f86..6eb08c56deea3ec5fe5cb3575987c6feaa0fb0b6 100644
GIT binary patch
literal 1143
zcmZ8g%Z{8d5WFMtDOe25V{_mXDThdrBIS@{<bj6qVr;M7WS05*RGW!5%F+F(uC8k9
zYqC|?x~@?NRhNZMyt<%cqOLorvuexYYK@yI%KIWEJw-O7K}+2nRihe&ZqmV(^KsGE
zJfyl+ZBAL!&t3Nd_zMcX^-(Xn>uPmWX9+m=YZL|tU<v%zTj{UW0i4QWSSpSLItZwj
zviM?k0${Vl=U4UF$JBL4RSUpqHDA|pB#D6`Xq*tmAokwox>GKku&ltI)Uhlk?<Y%D
zKrFM%T&5U$<FgW(%k3mYTXo3&W^gENWevk=&~BpWR~ZVXQ01#}ue!<gqT{QFl2-xC
z!a!1u+IA;Bo4ZNg{@!&yDlkTIZvl!9Q4dml({UdB-93jc&u3*gTlj>Msx|!3BQz5x
zfLb&M>p72a;Uch>VU2bd6;u~n4ZH;cp?at1^g{3Qs>L^F8Jf2WMDSRbr(Y*o(+94|
zxU?=(0TN{qbU*9K2zMBe18$IQp{hs`jJ5ub#blVIqh}A##KYz!72mgMA;-Iz7a0>a
zIm?U*1NI2tysHMyw6ei6jrs)q;1dhx#t($#M!`AKB<tXbX51`9iS)Ko+DEkc_@n7>
zAoWxz^=5{$mUM*T;u=s9+;<0vJV{@@q;Z^k-e3yMI!l(xB_FR|$3e&;h&}!Z3Uys2
zirIDu{gEHDKDh_%_a2E=#xdnpf%ad<S5u#=(O61bVuBrzH!sE%m;CjFMmB_Rk@OJo
zA$kT}`;5K~=)%X`nQj4c|DtypZ@hyU&@!JsKZN>+ytpMCS1a(~Toykm&JSAQKU9&e
M?okJPw!g~!0pJKm=l}o!

literal 332
zcmYk2OAf*?5JSC2;wVL=<+ou~iF;5!Qq(q}Gzf8dz@9)s)pTRe^Ej{P4juLw&|=H4
z#0(Xu#1s?!v;s9Qi1f*|pkr<$*OmJLN27chQY&Kw^3#Vd@`mF~8E}B7;;9#Ook)lz
ziQ#NtDM|>qp=Zux)JiPr$XE8<OH?~vnE3Z-%%`;VUz+%hwa6K|(OKJ^l6XO2L-jlx
V?QR^siNk6nwe-cYty1z^ya8pdFaH1l