update

app.py

@@ -86,10 +86,11 @@ def process_selected_models(uploaded_image, selected_display_names):
             try:
                 model, data_params = load_model(model_name)
                 pixel_values = pil_to_tensor(uploaded_image, **data_params).unsqueeze(0).to(device)
-                output = model(pixel_values)[0]
+                with torch.no_grad():
+                    output = model(pixel_values)[0]
                 reconstructed_image = tensor_to_pil(output[0].cpu(), **data_params)
 
-                del model
+                del model, pixel_values, output
                 torch.cuda.empty_cache()
 
                 result = gr.update(value=reconstructed_image, label=label)