Update app.py

app.py

@@ -0,0 +1,30 @@
+import gradio as gr
+from fastai.learner import load_learner
+import pandas as pd
+import numpy as np
+
+learn = load_learner('export.pkl')
+
+dados = pd.read_csv('valid.csv')
+ids = dados['user'].unique()
+
+def top_5_predictions(user):
+    items = pd.Series(learn.dls.classes['title']).unique()
+    
+    clas_items = dados.loc[dados['user'] == user, 'title']
+    
+    no_clas_items = np.setdiff1d(items, clas_items)
+
+    df = pd.DataFrame({'user': [user]*len(no_clas_items), 'title': no_clas_items})
+    
+    preds,_ = learn.get_preds(dl=learn.dls.test_dl(df))
+    
+    df['prediction'] = preds.numpy()
+    
+    top_5 = df.nlargest(5, 'prediction')
+    
+    return top_5['title'].tolist()
+
+iface = gr.Interface(fn=top_5_predictions, inputs=gr.inputs.Dropdown(choices=ids), outputs="text")
+
+iface.launch()