chroma

app.py

@@ -1,14 +1,16 @@
 import spaces
 import torch
 import gradio as gr
+import os
 from diffusers import FluxPipeline
 
 # Initialize model outside the function
 device = "cuda" if torch.cuda.is_available() else "cpu"
 dtype = torch.bfloat16
 file_url = "https://huggingface.co/lodestones/Chroma/blob/main/chroma-unlocked-v31.safetensors"
+huggingface_token = os.getenv("HUGGINGFACE_TOKEN")
 
-flux_pipeline = FluxPipeline.from_single_file(file_url, torch_dtype=dtype)
+flux_pipeline = FluxPipeline.from_single_file(file_url, torch_dtype=dtype, token=huggingface_token)
 flux_pipeline.to(device)
 
 @spaces.GPU()