Update app.py

app.py

@@ -1,6 +1,9 @@
 import torch
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 import gradio as gr
+from sklearn.ensemble import RandomForestClassifier
+from sklearn.feature_extraction.text import TfidfVectorizer
+
 import pickle
 
 vectorizer = pickle.load(open("tfidf.pickle", "rb"))