Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import streamlit as st
 #import pickle
-#import string
+import string
 from nltk.corpus import stopwords
 import nltk
 from nltk.stem.porter import PorterStemmer
@@ -33,8 +33,8 @@ def transform_text(text):
         
     return " ".join(y)
 
-tfidf = pickle.load(open('vectorizer.pkl','rb'))
-model = pickle.load(open('model.pkl','rb'))
+tfidf = load_learner('vectorizer.pkl','rb')
+model = load_learner('model.pkl','rb')
 
 
 def predict_spam(input_sms):