Update my_model/utilities.py

my_model/utilities.py

@@ -298,14 +298,13 @@ def free_gpu_resources():
     """
     try:
         if torch.cuda.is_available():
-            torch.cuda.empty_cache()
-            gc.collect()
-            torch.cuda.empty_cache()
-            gc.collect()
-
-            st.write("GPU memory has been cleared.")
-    except Exception as e:
-        st.write(f"Error occurred while clearing GPU memory")
+        torch.cuda.empty_cache()
+        torch.cuda.empty_cache()
+        gc.collect()
+        gc.collect()
+
+        st.write("GPU memory has been cleared.")
+