Spaces:
Sleeping
Sleeping
import modal | |
from src.gradio_interface import gradio_interface | |
from config.config import prompts, models # Indirect import | |
# Define the Modal image | |
image = ( | |
#modal.Image.from_registry("nvidia/cuda:12.2.0-devel-ubuntu22.04", add_python="3.9") | |
modal.Image.debian_slim(python_version="3.9") # Base image | |
.apt_install( | |
"git", | |
) | |
.pip_install( | |
"diffusers", | |
"transformers", | |
"torch", | |
"accelerate", | |
"gradio>=4.44.1", | |
"safetensors", | |
"pillow", | |
"sentencepiece", | |
f"git+https://github.com/huggingface/transformers.git" | |
) | |
.env( | |
{ | |
"HF_HUB_ENABLE_HF_TRANSFER": "1", "HF_HOME": "HF_HOME" | |
} | |
) | |
) | |
# Create the Modal app | |
app = modal.App("ctb-image-generator-modal", image=image) | |
with image.imports(): | |
import diffusers | |
import os | |
import gradio | |
import torch | |
import sentencepiece | |
import importlib | |
print("Modal app created.") | |
# Entry point for local execution | |
def main(): | |
print("Launching Gradio interface...") | |
# demo.launch() | |
gradio_interface.launch() | |