File size: 821 Bytes
8742d2f 7eee0d3 6a96924 b16d5bd 4143951 45650be 8742d2f 4143951 |
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 |
import gradio as gr
import surprise
from surprise import dump
file_name = 'modelo_entrenado.sav'
model = dump.load(file_name)[1]
def predict_rating(userId, id):
prediction = model.predict(userId, id)
if prediction.est >= 3.5:
mensaje = "Es hora de verla!!", prediction.est
elif prediction.est >= 2.5 and prediction.est < 3.5:
mensaje = "Puedes verla luego!", prediction.est
else:
mensaje = "No es la película que esta buscando!", prediction.est
return mensaje
iface = gr.Interface(fn=predict_rating,
inputs= [gr.inputs.Textbox(lines=1,placeholder="ingrese su número de usuario aquí"),
gr.inputs.Textbox(lines=1,placeholder="ingrese su número de usuario aquí")],
outputs="text")
iface.launch() |