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