Update app.py

app.py

@@ -43,7 +43,7 @@ def PredictNCM(txt):
     X = np.array(tk.texts_to_sequences([_+(120-len(_))*" " for _ in x]))
     pred = model.predict(X, verbose=0)[0]
     aux = np.argsort(pred)[::-1][:5]
-    return {str(valid_ncms[i]):float(pred[i]) for i in aux}, ncm_table.loc[valid_ncms[aux[0]],"DESCRICAO"]
+    return {f"{int(valid_ncms[i]):08}":float(pred[i]) for i in aux}, ncm_table.loc[valid_ncms[aux[0]],"DESCRICAO"]
 
 
 demo = gr.Interface(fn=PredictNCM, outputs=[components.Label(label="NCMs"), components.Textbox(label="Descrição do NCM")], title='AFRAC NOTA CERTA',