File size: 545 Bytes
bdf9962
 
 
 
 
fa851d1
bdf9962
 
 
fa851d1
bdf9962
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
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=["/"])