app

app.py

@@ -13,9 +13,9 @@ def mlm(image, text):
         # 'location': f'[CLS] The {cls_name} normally appears at or near the [MASK] of a cell. [SEP]',
         # 'color': f'[CLS] When a cell is histologically stained, the {cls_name} are in [MASK] color. [SEP]',
         # 'shape': f'[CLS] Mostly the shape of {cls_name} is [MASK]. [SEP]',
-        'location': f'[CLS] The location of {text} is at [MASK] . [SEP]',
-        'color': f'[CLS] The typical color of {text} is [MASK] . [SEP]',
-        'shape': f'[CLS] The typical shape of {text} is [MASK] . [SEP]',
+        'location': f'[MASK] is the typical location where {text} is at.',
+        'color': f'[MASK] is the typical color of {text}.',
+        'shape': f'[MASK] is the typical shape of {text}.',
         #'def': f'{cls_name} is a  . [SEP]',
     }
     ans = list()