Fix moving code

app.py

@@ -210,8 +210,6 @@ def restore_in_Xmin(
 
     denoise_image = HWC3(np.array(denoise_image))
 
-    torch.cuda.set_device(SUPIR_device)
-
     if model_select != model.current_model:
         print('load ' + model_select)
         if model_select == 'v0-Q':
@@ -220,9 +218,6 @@ def restore_in_Xmin(
             model.load_state_dict(ckpt_F, strict=False)
         model.current_model = model_select
 
-    model.ae_dtype = convert_dtype(ae_dtype)
-    model.model.dtype = convert_dtype(diff_dtype)
-
     # Allocation
     if allocation == 1:
         return restore_in_1min(
@@ -338,6 +333,11 @@ def restore(
     start = time.time()
     print('restore ==>>')
 
+    torch.cuda.set_device(SUPIR_device)
+
+    model.ae_dtype = convert_dtype(ae_dtype)
+    model.model.dtype = convert_dtype(diff_dtype)
+
     input_image = upscale_image(input_image, upscale, unit_resolution=32, min_size=min_size)
     LQ = np.array(input_image) / 255.0
     LQ = np.power(LQ, gamma_correction)
@@ -378,9 +378,6 @@ def restore(
     " pixels large and " + str(result_height) + \
     " pixels high, so a resolution of " + f'{result_width * result_height:,}' + " pixels."
     print(information)
-    unique_name = str(uuid.uuid4()) + ".png"
-    results[0].save(unique_name)
-    print(unique_name)
 
     # Only one image can be shown in the slider
     return [noisy_image] + [results[0]], gr.update(label="Downloadable results in *." + output_format + " format", format = output_format, value = results), gr.update(value = information, visible = True)