Update api_server.py

api_server.py

@@ -75,12 +75,30 @@ def get_jpg_files(path):
 # image_folder = '/content/drive/MyDrive/chiikawa'  # 替換成你的目錄路徑
 # jpg_files = get_jpg_files(image_folder)
 
+import psutil
 
+def check_memory_usage():
+    # Get memory details
+    memory_info = psutil.virtual_memory()
+    
+    total_memory = memory_info.total / (1024 * 1024)  # Convert bytes to MB
+    available_memory = memory_info.available / (1024 * 1024)
+    used_memory = memory_info.used / (1024 * 1024)
+    memory_usage_percent = memory_info.percent
+    
+    print(f"Total Memory: {total_memory:.2f} MB")
+    print(f"Available Memory: {available_memory:.2f} MB")
+    print(f"Used Memory: {used_memory:.2f} MB")
+    print(f"Memory Usage (%): {memory_usage_percent}%")
+
+# Run the function
+check_memory_usage()
 # Initialize the Flask application
 app = Flask(__name__)
 # Initialize the ClipModel at the start
 clip_model = ClipModel()
 
+check_memory_usage()
 
 # API route for prediction(YOLO)
 @app.route('/predict', methods=['POST'])