oscarvillafuerte commited on
Commit
b740ceb
·
verified ·
1 Parent(s): 83d5e5c

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +36 -0
app.py ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ import tensorflow as tf
3
+ from tensorflow.keras.datasets import imdb
4
+ from tensorflow.keras.preprocessing.sequence import pad_sequences
5
+
6
+ word_index = imdb.get_word_index()
7
+ maximo_num_palabras = 20000
8
+
9
+ model = tf.keras.models.load_model("opiniones.h5")
10
+
11
+ def reviewnueva(review, word_index, maximo_num_palabras):
12
+ sequence = []
13
+ for word in review.split():
14
+ index = word_index.get(word.lower(), 0)
15
+ if index < maximo_num_palabras:
16
+ sequence.append(index)
17
+ return sequence
18
+
19
+ def predict_sentimiento(review):
20
+ sequence = reviewnueva(review, word_index)
21
+ prediccion = model.predict(sequence)
22
+ if prediccion[0][0]>=0.5 :
23
+ sentimiento = "Positivo"
24
+ else:
25
+ sentimiento = "Negativo"
26
+ return sentimiento
27
+
28
+ st.title("Ingrese una review para poder calficarla como positiva o negativ")
29
+ review = st.text_area("Ingrese la reseña aqui", height= 200)
30
+
31
+ if st.button("Predecir el sentimiento de la reseña")
32
+ if review:
33
+ sentimiento = predict_sentimiento(review)
34
+ st.write(f'El sentimiento es :{sentimiento}')
35
+ else:
36
+ st.write(f'Ingrese una review')