Update app.py

app.py

@@ -28,8 +28,8 @@ def load_cnn():
   model.eval()
 
   return model
-def predict_sentiment(text, model, vocab, tokenizer):
-    if torch == True:
+def predict_sentiment(text, model, vocab, tokenizer, torch_text = False):
+    if torch_text == True:
         processor.transform(text)
         tokens = tokenizer(text)
         encoded = [vocab[token] for token in tokens]
@@ -75,7 +75,7 @@ with st.expander("Model 2: CNN Sentiment analysis"):
     user_input = st.text_area("Enter text here...", key='model2_input')
     if st.button('Analyze', key='model2_button'):
         # Displaying output
-        result = predict_sentiment(user_input, model_2, torch=True)
+        result = predict_sentiment(user_input, model_2, torch_text=True)
         if result >= 0.5:
             st.write('The sentiment is: Positive 😀', key='model2_poswrite')
         else: