Spaces:
Sleeping
Sleeping
File size: 1,973 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 66 67 68 69 70 |
import modal
# Print debug information
print("Importing Modal and setting up the app...")
# Define a custom image with Python and some dependencies
print("Building custom image...")
image = (
modal.Image.debian_slim(python_version="3.11") # Base image
.pip_install(
"numpy",
"pandas",
"diffusers",
"transformers",
"torch",
"accelerate",
"gradio",
"safetensors",
"pillow",
) # Install Python packages
.run_commands("echo 'Image build complete!'") # Run a shell command
)
# Define a function to run inside the container
#@app.function(image=image)
# Define the Modal app
app = modal.App("functions-app")
@app.function()
def message_func (message = "default"):
print("message function")
new_message = message + " ok, it works!"
return new_message
@app.local_entrypoint()
def main():
# Import libraries and print their versions
# import numpy as np
# import pandas as pd
# import torch
# import diffusers
# import transformers
# import gradio as gr
# from PIL import Image as PILImage
# print("def main function")
# print("Hello from Modal!")
# print("NumPy version:", np.__version__)
# print("Pandas version:", pd.__version__)
# print("PyTorch version:", torch.__version__)
# print("Diffusers version:", diffusers.__version__) # Corrected: Use the library's __version__
# print("Transformers version:", transformers.__version__) # Corrected: Use the library's __version__
# print("Gradio version:", gr.__version__)
# print("Pillow version:", PILImage.__version__)
remote_message = "remote message!"
local_message = "local message"
message_func.remote(remote_message)
message_func.local(local_message)
# # # Run the function locally (for testing)
# if __name__ == "__main__":
# print("Running the function locally...")
# main.local()
# main.remote()
|