Spaces:
Sleeping
Sleeping
| # img_gen.py | |
| #img_gen_modal.py | |
| import modal | |
| import os | |
| import shutil | |
| # Define the Modal image | |
| image = ( | |
| modal.Image.debian_slim(python_version="3.9") | |
| ) | |
| # Create a Modal app | |
| app = modal.App("img-see-vol-data", image=image) | |
| flux_model_vol = modal.Volume.from_name("flux-model-vol",create_if_missing=False) # Reference your volume | |
| def main(): | |
| # Define where to store the model | |
| download_model_name = "black-forest-labs/FLUX.1-dev" # e.g., "stabilityai/stable-diffusion-2" | |
| local_dir = "data/FLUX.1-dev" | |
| model_path = "/data" | |
| if os.path.exists(model_path): | |
| print(f"Contents of {model_path}:") | |
| print(os.listdir(model_path)) | |
| else: | |
| print(f"Model path {model_path} does not exist.") | |