import torch
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__":
    # Set the model path
    model_path = "deepseek-ai/Janus-1.3B"

    # Load the model and processor
    vl_gpt, vl_chat_processor = load_model_and_processor(model_path)

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