Add bidirectionalllamaattention to imports

app.py

@@ -8,7 +8,7 @@ from llama_diffusion_model import disable_dropout
 import os
 import importlib
 from huggingface_hub import hf_hub_download
-from llama_diffusion_model import CustomTransformerModel, CustomTransformerConfig, disable_dropout
+from llama_diffusion_model import CustomTransformerModel, CustomTransformerConfig, BidirectionalLlamaAttention, disable_dropout
 
 hf_token = os.getenv("HF_TOKEN")