Spaces:
Sleeping
Sleeping
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=["/"]) |