Spaces:
Build error
Build error
| import gradio as gr | |
| from joblib import load | |
| # Load the vectorizer and model | |
| vectorizer = load('tfidf_vectorizer.pkl') | |
| model = load('fake_review_detector.pkl') | |
| # Prediction function | |
| def predict_review(review): | |
| # Transform the input text into TF-IDF | |
| review_tfidf = vectorizer.transform([review]) | |
| # Make a prediction | |
| prediction = model.predict(review_tfidf)[0] | |
| # Return the result | |
| return "THIS IS NOT FAKE" if prediction == 0 else "THIS IS FAKE" | |
| # Gradio interface | |
| interface = gr.Interface( | |
| fn=predict_review, | |
| inputs="text", | |
| outputs="text", | |
| title="Fake Review Detector", | |
| description="Type a review below to determine if it is fake or not." | |
| ) | |
| # Launch the interface | |
| interface.launch() | |