Update app.py

app.py

@@ -1,3 +1,4 @@
+import spaces
 import os
 import requests
 import torch
@@ -7,7 +8,6 @@ from PIL import Image
 from RealESRGAN import RealESRGAN
 import cv2
 import numpy as np
-import spaces
 
 # Constants
 SD15_WEIGHTS = "weights"
@@ -39,6 +39,13 @@ def download_file(url, folder_path, filename):
 
 # Download necessary models and files
 
+@spaces.GPU
+def gradio_process_image(input_image, resolution, num_inference_steps, strength, hdr, guidance_scale):
+    prompt = "masterpiece, best quality, highres"
+    negative_prompt = "low quality, normal quality, ugly, blurry, blur, lowres, bad anatomy, bad hands, cropped, worst quality, verybadimagenegative_v1.3, JuggernautNegative-neg"
+    result = process_image(input_image, prompt, negative_prompt, resolution, num_inference_steps, guidance_scale, strength, hdr)
+    return result
+    
 # MODEL
 download_file(
     "https://huggingface.co/dantea1118/juggernaut_reborn/resolve/main/juggernaut_reborn.safetensors?download=true",
@@ -216,12 +223,7 @@ def process_image(input_image, prompt, negative_prompt, resolution=2048, num_inf
 
     return result
 
-@spaces.GPU
-def gradio_process_image(input_image, resolution, num_inference_steps, strength, hdr, guidance_scale):
-    prompt = "masterpiece, best quality, highres"
-    negative_prompt = "low quality, normal quality, ugly, blurry, blur, lowres, bad anatomy, bad hands, cropped, worst quality, verybadimagenegative_v1.3, JuggernautNegative-neg"
-    result = process_image(input_image, prompt, negative_prompt, resolution, num_inference_steps, guidance_scale, strength, hdr)
-    return result
+
 
 # Simple options
 simple_options = [