Spaces:
Sleeping
Sleeping
File size: 1,981 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 53 54 55 56 57 58 59 60 61 62 63 64 65 |
import modal
from src.gradio_interface_modal import gradio_interface_modal
from config.config import prompts, models # Indirect import
# Define the Modal image
image = (
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)
print("Modal app created.")
# Entry point for local execution
@app.function(secrets=[modal.Secret.from_name("huggingface-token")])
def main():
with modal.enable_output():
print("Hello from ctb_modal!")
print("Running debug check...")
# Debug function to check installed packages
def check_dependencies():
packages = [
"diffusers", # For Stable Diffusion
"transformers", # For Hugging Face models
"torch", # PyTorch
"accelerate", # For distributed training/inference
"gradio", # For the Gradio interface (updated to latest version)
"safetensors", # For safe model loading
"pillow", # For image processing
"sentencepiece"
]
for package in packages:
try:
import importlib
module = importlib.import_module(package)
print(f" {package} is installed. Version:")
except ImportError:
print(f" {package} is NOT installed.")
check_dependencies()
print("Launching Gradio interface...")
# demo.launch()
gradio_interface_modal()
|