Update clip_model.py

clip_model.py

@@ -2,6 +2,7 @@ import os
 import torch
 from PIL import Image
 from transformers import ChineseCLIPProcessor, ChineseCLIPModel
+import psutil
 
 def check_memory_usage():
     # Get memory details