import spaces
from model_loader import load_model_and_processor
from image_generator import process_and_generate
from gradio_interface import create_gradio_interface

if __name__ == "__main__":
    import subprocess
    subprocess.run('pip install "flash-attn>=2.1.0" --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)




    # Create and launch the Gradio interface
    demo = create_gradio_interface(process_and_generate)
    demo.launch(allowed_paths=["/"])