app.py

import spaces
import torch
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 --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=["/"])