File size: 1,156 Bytes
0d5600b
 
 
ab6f726
0d5600b
 
 
 
 
 
 
 
 
 
 
 
 
2d7a0af
0d5600b
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
34
35
36
37
38
39
import streamlit as st
import tensorflow as tf
from tensorflow.keras.datasets import imdb
from tensorflow.keras.preprocessing.sequence import pad_sequences
import numpy as np

word_index = imdb.get_word_index()
maximo_num_palabras = 20000

def reviewnueva(review, word_index, maxmimo_num_palabras):
  sequence = []
  for word in review.split():
    index = word_index.get(word.lower(),0)
    if index < maximo_num_palabras:
      sequence.append(index)
      return sequence

model = tf.keras.models.load_model('opiniones.h5')

def predict_sentiento(review):
    sequence = reviewnueva(review, word_index)
    
    sentimiento = model.predict(nuevareviewpad)
    if(sentimiento[0][0] > 0.5):
      print("El sentimiento es positiva")
    else:
      print("El sentimiento es negativa")
    return sentimiento

st.title("Ingrese una review para poder calificarla como positiva o negativa")
review = st.text_area("Ingrese reseña aquí ", height=200)


if st.button("Predecir sentimiento"):
    if review:
        sentimiento = predict_sentiento(review)
        st.write(f"El sentimiento es: {sentimiento}")
    else:
        st.write(f"Ingrese una review.")