Update my_model/utilities.py

my_model/utilities.py

@@ -9,6 +9,7 @@ import matplotlib.pyplot as plt
 from IPython import get_ipython
 import sys
 import gc
+import streamlit as st
 
 
 class VQADataProcessor:
@@ -302,9 +303,9 @@ def free_gpu_resources():
             torch.cuda.empty_cache()
             gc.collect()
 
-            print("GPU memory has been cleared.")
+            st.write("GPU memory has been cleared.")
     except Exception as e:
-        print(f"Error occurred while clearing GPU memory: {str(e)}")
+        st.write(f"Error occurred while clearing GPU memory")