app modified

app.py

@@ -440,7 +440,7 @@ class ModeloDataset:
         i=0
         out=[]
         for iden in labels:
-          #if i<len(ids):
+          
            if etiquetas:
               out.append(self.salida_texto( iden,np.array(tokens[i])))
            else:   
@@ -448,13 +448,16 @@ class ModeloDataset:
            i=i+1
         
         return out   
-    def salida_texto2_es(self, ids,pre_tokens):
+    def salida_texto2_es(self, ids,pre_tokens,etiquetas):
         i=0
         out=[]
         for iden in pre_tokens:
           if i<len(ids):
-           
-           out.append(self.salida_texto_es( ids[i],np.array(pre_tokens[i])) )
+           if etiquetas:
+               out.append(self.salida_texto_es( ids[i],np.array(pre_tokens[i])) )
+           else:
+               out.append(self.salida_texto_es( ids[i],self.reemplazo_fake(np.array(pre_tokens[i]))))
+                 
            i=i+1
         
         return out    
@@ -555,12 +558,12 @@ def procesar(texto,archivo, etiquetas):
 
                 if model.idioma=="es":
                   
-                  out=modelo.salida_texto2_es( ides,predicted)
-                  print('out:',out)
+                  out=modelo.salida_texto2_es( ides,predicted,etiquetas)
+                  print('out es:',out)
                   df_new[item] = modelo.unir_array(out)
                 else:  
                   out=modelo.salida_texto2( ides,predicted,etiquetas)#tokens,labels 
-                  print('out:',out)
+                  print('out en:',out)
                   df_new[item] = modelo.unir_array(out)
             return "", df_new, df_new.to_csv(sep='\t', encoding='utf-8',index=False)
             #return "", df_new, df_new.to_excel( index=False)