fix

app.py

@@ -66,6 +66,6 @@ if __name__=='__main__':
     input_ids = torch.tensor([input_ids_1,input_ids_2])
 
     outputs = SkeletonAlbertForMaskedLM(model,input_ids,interventions = {0:{'lay':[(8,1,[0,1])]}})
-    logprobs = F.log_softmax(outputs.logits, dim = -1)
+    logprobs = F.log_softmax(outputs['logits'], dim = -1)
     preds = [torch.multinomial(torch.exp(probs), num_samples=1).squeeze(dim=-1) for probs in logprobs[0]]
     st.write([tokenizer.decode([token]) for token in preds])