thomasgauthier's picture
first commit
bdf9962
raw
history blame
545 Bytes
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=["/"])