to_cpu

app.py

@@ -346,8 +346,8 @@ class Experiment(Component):
         # @spaces.GPU
         def render_plots(data_id, *metric_inputs):
             # Clear Cache Files
-            print(f"GPU Check: {torch.cuda.is_available()}")
-            print("Which GPU: ", torch.cuda.current_device())
+            # print(f"GPU Check: {torch.cuda.is_available()}")
+            # print("Which GPU: ", torch.cuda.current_device())
             cache_dir = f"{os.environ['GRADIO_TEMP_DIR']}/res"
             if not os.path.exists(cache_dir): os.makedirs(cache_dir)
             for f in os.listdir(cache_dir):