bring back some

app.py

@@ -163,11 +163,11 @@ def run_intervention(interventions,batch_size,model,masked_ids_option_1,masked_i
 
 if __name__=='__main__':
     wide_setup()
-    #load_css('style.css')
-    #tokenizer,model = load_model()
-    #num_layers, num_heads = model.config.num_hidden_layers, model.config.num_attention_heads
-    #st.write(num_layers,num_heads)
-    #mask_id = tokenizer('[MASK]').input_ids[1:-1][0]
+    load_css('style.css')
+    tokenizer,model = load_model()
+    num_layers, num_heads = model.config.num_hidden_layers, model.config.num_attention_heads
+    st.write(num_layers,num_heads)
+    mask_id = tokenizer('[MASK]').input_ids[1:-1][0]
 
     main_area = st.empty()
 

style.css

@@ -0,0 +1,4 @@
+div.stButton > button:first-child {
+  color: #4f8bf9;
+  font-size: 0.5rem;
+}