Update app.py

app.py

@@ -4,6 +4,9 @@ import pandas as pd
 import numpy as np
 import torch
 
+from preprocess import Preprocess
+from utility import Utility
+
 st.title("Movie Genre Predictor")
 
 st.subheader("Enter the text you'd like to analyze.")
@@ -23,8 +26,8 @@ if st.button("Predict"):
     utility = Utility()
     id2label, label2id, tokenizer, tokenized_plot = utility.tokenize(clean_plot, ["Action","Drama", "Romance", "Comedy", "Thriller"])
     st.write(tokenized_plot)
-    xtrain_input_ids = [np.asarray(tokenized_plot['input_ids'])]
-    xtrain_attention_mask = [np.asarray(tokenized_plot['attention_mask'])]
+    input_ids = [np.asarray(tokenized_plot['input_ids'])]
+    attention_mask = [np.asarray(tokenized_plot['attention_mask'])]
     st.write([np.asarray(tokenized_plot['input_ids'])])
     st.write(clean_plot)