File size: 438 Bytes
e1d6915
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
# modal_app.py
import modal
#IMPORT gradio_interface
from src.gradio_interface import demo

# Create a Modal app
app = modal.App("ctb-image-generator")

# Define the Modal image
image = modal.Image.debian_slim().pip_install(
    "diffusers", 
    "transformers",
    "torch",
    "accelerate",
    "gradio"
)

@app.local_entrypoint()
def main():
    with modal.enable_output():
        demo.launch()

if __name__ == "__main__":
    main()