Fix downscale

app.py

@@ -304,45 +304,45 @@ def restore_in_Xmin(
 
 @spaces.GPU(duration=59)
 def restore_in_1min(*args, **kwargs):
-    return restore(*args, **kwargs)
+    return restore_on_gpu(*args, **kwargs)
 
 @spaces.GPU(duration=119)
 def restore_in_2min(*args, **kwargs):
-    return restore(*args, **kwargs)
+    return restore_on_gpu(*args, **kwargs)
 
 @spaces.GPU(duration=179)
 def restore_in_3min(*args, **kwargs):
-    return restore(*args, **kwargs)
+    return restore_on_gpu(*args, **kwargs)
 
 @spaces.GPU(duration=239)
 def restore_in_4min(*args, **kwargs):
-    return restore(*args, **kwargs)
+    return restore_on_gpu(*args, **kwargs)
 
 @spaces.GPU(duration=299)
 def restore_in_5min(*args, **kwargs):
-    return restore(*args, **kwargs)
+    return restore_on_gpu(*args, **kwargs)
 
 @spaces.GPU(duration=359)
 def restore_in_6min(*args, **kwargs):
-    return restore(*args, **kwargs)
+    return restore_on_gpu(*args, **kwargs)
 
 @spaces.GPU(duration=419)
 def restore_in_7min(*args, **kwargs):
-    return restore(*args, **kwargs)
+    return restore_on_gpu(*args, **kwargs)
 
 @spaces.GPU(duration=479)
 def restore_in_8min(*args, **kwargs):
-    return restore(*args, **kwargs)
+    return restore_on_gpu(*args, **kwargs)
 
 @spaces.GPU(duration=539)
 def restore_in_9min(*args, **kwargs):
-    return restore(*args, **kwargs)
+    return restore_on_gpu(*args, **kwargs)
 
 @spaces.GPU(duration=599)
 def restore_in_10min(*args, **kwargs):
-    return restore(*args, **kwargs)
+    return restore_on_gpu(*args, **kwargs)
 
-def restore(
+def restore_on_gpu(
     noisy_image,
     input_image,
     prompt,