Spaces:

AFRAC
/

NCM_DEMO

Running

mfilipak commited on Jul 7, 2023

Commit

c5d0c16

1 Parent(s): 4bdd502

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -27,7 +27,7 @@ def PredictNCM(txt):
     print(txt)
     X = np.array(tk.texts_to_sequences([_+(120-len(_))*" " for _ in x]))
-    pred = model.predict(X)[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"]

     print(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"]