Spaces:
Sleeping
Sleeping
File size: 915 Bytes
990dc13 07d5446 990dc13 07d5446 990dc13 07d5446 990dc13 07d5446 990dc13 b1ce276 2cdf71f 18f2d2a 990dc13 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 |
import numpy as np
import os
import gradio as gr
import xgboost as xgb
from sklearn.feature_extraction.text import TfidfVectorizer
os.environ["WANDB_DISABLED"] = "true"
model_file_name = "xgb_reg.pkl"
vectorizer_file_name = 'vectorizer.pk'
#load
xgb_model_loaded = pickle.load(open(model_file_name, "rb"))
vectorizer_loaded = pickle.load(open(vectorizer_file_name, "rb"))
# predict
def predict_sentiment(predict_texts):
predictions_loaded = xgb_model_loaded.predict(
vectorizer_loaded.transform(predict_texts))
return vectorizer_loaded
#le.inverse_transform(predictions_loaded)
interface = gr.Interface(
fn=predict_sentiment,
inputs='text',
outputs=['text'],
title='Croatian Movie reviews Sentiment Analysis',
examples= ["Volim kavu","Ne volim kavu"],
description='Get the positive/neutral/negative sentiment for the given input.'
)
interface.launch(inline = False)
|