Update app.py

app.py

@@ -3,6 +3,8 @@ import gradio as gr
 
 learn = load_learner('export (1).pkl')
 
+labels = learn.dls.vocab
+
 def predict(frase):
     pred, pred_idx, probs = learn.predict(frase)
     return {labels[i]: float(probs[i]) for i in range(len(labels))}