Update app.py

app.py

@@ -1,3 +1,5 @@
+import os
+os.system('pip install --upgrade --pre --extra-index-url https://download.pytorch.org/whl/nightly/cu126 "torch<2.9" spaces')
 import gradio as gr
 import numpy as np
 import spaces
@@ -9,7 +11,9 @@ from PIL import Image
 from diffusers import FluxKontextPipeline
 from diffusers.utils import load_image, peft_utils
 from huggingface_hub import hf_hub_download, HfFileSystem, ModelCard
+from huggingface_hub import snapshot_download
 from safetensors.torch import load_file
+import tempfile
 import requests
 import re