Update app.py

app.py

@@ -1,4 +1,6 @@
-import spaces
+import subprocess
+subprocess.run(['sh', './spaces.sh'])
+
 import os
 
 os.putenv('PYTORCH_NVML_BASED_CUDA_CHECK','1')
@@ -11,6 +13,8 @@ os.environ['PYTORCH_CUDA_ALLOC_CONF'] = ','.join(alloc_conf_parts)
 os.environ["SAFETENSORS_FAST_GPU"] = "1"
 os.putenv('HF_HUB_ENABLE_HF_TRANSFER','1')
 
+import spaces
+
 import gradio as gr
 import numpy as np
 import random
@@ -93,7 +97,7 @@ pipe = StableDiffusion3Pipeline.from_pretrained(
     transformer=None,
     tokenizer_3=T5TokenizerFast.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", add_prefix_space=False, use_fast=True, subfolder="tokenizer_3", token=True),
     #torch_dtype=torch.bfloat16,
-    #use_safetensors=False,
+    use_safetensors=True,
 )
 text_encoder=CLIPTextModelWithProjection.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", subfolder='text_encoder', token=True).to(torch.device("cuda:0"), dtype=torch.bfloat16)
 text_encoder_2=CLIPTextModelWithProjection.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", subfolder='text_encoder_2',token=True).to(torch.device("cuda:0"), dtype=torch.bfloat16)