adjusting mask

pages/fill_mask.py

@@ -50,12 +50,11 @@ def app():
             model = 'bert-base-uncased'
 
         with st.spinner('Loading Model... (This may take a while)'):
-            # unmasker = get_model()    
+            unmasker = get_model()    
             st.success('Model loaded correctly!')
         
             gen = st.info('Generating text...')
-            # answer = unmasker(prompt)     
-            answer = [{'sequence': "[CLS] hello i'm a fashion model. [SEP]",  'score': 0.1073106899857521,  'token': 4827,  'token_str': 'fashion'}, {'sequence': "[CLS] hello i'm a role model. [SEP]",  'score': 0.08774490654468536,  'token': 2535,  'token_str': 'role'}, {'sequence': "[CLS] hello i'm a new model. [SEP]",  'score': 0.05338378623127937,  'token': 2047,  'token_str': 'new'}, {'sequence': "[CLS] hello i'm a super model. [SEP]",  'score': 0.04667217284440994,  'token': 3565,  'token_str': 'super'}, {'sequence': "[CLS] hello i'm a fine model. [SEP]",  'score': 0.027095865458250046, 'token': 2986,  'token_str': 'fine'}]                 
+            answer = unmasker(prompt)     
             gen.empty()    
 
     with col1: