Sambhavnoobcoder's picture
changed gradio interface
a7f2f89
raw
history blame
1.19 kB
import tensorflow as tf
import gradio as gr
import numpy as np
# Load the saved model
model = load_model('sentimentality.h5')
# Define the function to preprocess the text
def preprocess(text):
# Tokenize the text
tokenizer = tf.keras.preprocessing.text.Tokenizer(num_words=10000)
tokenizer.fit_on_texts(text)
sequences = tokenizer.texts_to_sequences(text)
# Pad the sequences to a fixed length of 30
padded_sequences = tf.keras.preprocessing.sequence.pad_sequences(sequences, maxlen=30, padding='post')
return np.array(padded_sequences)
# Define the function to make predictions on the text
def sentiment_analysis(text):
preprocessed_text = preprocess([text])
prediction = model.predict(preprocessed_text)[0][0]
if prediction >= 0.5:
return "positive"
else:
return "negative"
# Define the Gradio interface
interface = gr.Interface(
fn=get_sentiment_scores,
inputs=gr.inputs.Textbox(placeholder="Enter a positive or negative sentence here..."),
outputs=gr.outputs.Textbox(label="Sentiment Label"),
examples=[["This is wonderful!"], ["I hate this product."]]
)
# Launch the interface
interface.launch()