Update app.py

app.py

@@ -1,4 +1,9 @@
 import spaces
+import os
+
+os.environ["SAFETENSORS_FAST_GPU"] = "1"
+os.putenv("HF_HUB_ENABLE_HF_TRANSFER","1")
+
 import gradio as gr
 import numpy as np
 import random
@@ -12,7 +17,6 @@ import re
 import paramiko
 import urllib
 import time
-import os
 from image_gen_aux import UpscaleWithModel
 from huggingface_hub import hf_hub_download
 import datetime
@@ -34,8 +38,6 @@ torch.backends.cudnn.benchmark = False
 #torch.backends.cuda.preferred_linalg_library="cusolver"
 
 hftoken = os.getenv("HF_AUTH_TOKEN") 
-os.environ["SAFETENSORS_FAST_GPU"] = "1"
-os.putenv("HF_HUB_ENABLE_HF_TRANSFER","1")
 
 code = r'''
 import torch