app

app.py

@@ -41,6 +41,8 @@ def mlm(image, text):
             #print(predicted_index)
             for idx in predicted_index:
                 predicted_tokens.append(tokenizer.convert_ids_to_tokens([idx])[0])
+            for i in range(2):
+                res[text][k].append(predicted_tokens)
 
         return image, res