Update app.py

app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-from functions_preprocess import LinguisticPreprocessor, download_if_non_existent
+from functions_preprocess import LinguisticPreprocessor, download_if_non_existent, CNN
 import pickle
 import nltk
 nltk.download('stopwords')
@@ -22,6 +22,7 @@ def load_model():
   return model
 
 def load_cnn():
+  model = CNN(len(vocab), 300, 128, [3, 8], 0.5, 2)
   model.load_state_dict(torch.load('model_cnn.pkl'))
   model.eval()