import gradio as gr
from sklearn.feature_extraction.text import CountVectorizer
import joblib

vectorizer = CountVectorizer()
nb_classifier = joblib.load('./nb_classifier.pkl')
def classify(text):
  corpus=[text]
  features = vectorizer.transform(corpus)
  features = features.toarray()
  prediction = nb_classifier.predict(features)
  if(prediction == 1):
     return "Fake News"
  else:
     return "Not Fake News"

GUI = gr.Interface(inputs = ['text'], outputs = ['text'], fn = classify, title = "Fake News Detection System")
GUI.launch()