Spaces:
Sleeping
Sleeping
File size: 1,121 Bytes
4f48282 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 |
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
@app.local_entrypoint()
def main():
print("Launching Gradio interface...")
# demo.launch()
gradio_interface.launch()
|