Update app.py

app.py

@@ -27,7 +27,7 @@ SAVE_DIR = path_dict['gradio_save_dir']
 os.environ["GRADIO_TEMP_DIR"] = path_dict['gradio_temp_dir']
 os.environ['TMPDIR'] = path_dict['gradio_temp_dir']
 
-
+@spaces.GPU
 # Function of loading pre-trained models.
 def load_models():
     base_model_path = snapshot_download("black-forest-labs/FLUX.1-Fill-dev")
@@ -40,7 +40,6 @@ def load_models():
     model = Calligrapher(pipe, image_encoder_path, calligrapher_path, device="cuda", num_tokens=128)
     return model
 
-@spaces.GPU
 # Init models.
 model = load_models()
 print('Model loaded!')