importmatplotlib.pyplotaspltimportnumpyasnpimportpandasaspdimporttorchfromglobimportglobfromtqdmimporttqdmfrommodelimportSamOutimportpolarsasplfromcollectionsimportCounterdeftrain():voc=pd.read_pickle("tot