Update app.py

app.py

@@ -3,6 +3,7 @@ import gradio as gr
 import json
 import logging
 import torch
+from os import path
 from PIL import Image
 import spaces
 from diffusers import DiffusionPipeline, AutoPipelineForText2Image
@@ -10,7 +11,9 @@ from diffusers import StableDiffusion3Pipeline, FlowMatchEulerDiscreteScheduler,
 import copy
 import random
 import time
-from huggingface_hub import login
+from huggingface_hub import login, hf_hub_download
+import safetensors.torch
+from safetensors.torch import load_file
 hf_token = os.environ.get("HF_TOKEN")
 login(token=hf_token)