Update app.py

app.py

@@ -253,13 +253,13 @@
 import logging
 import random
 import warnings
-import os
+import os,shutil,subprocess
 import torch
 import numpy as np
 from diffusers import FluxControlNetModel
 from diffusers.pipelines import FluxControlNetPipeline
 from PIL import Image
-from huggingface_hub import snapshot_download,login,HfFolder
+from huggingface_hub import snapshot_download,login
 import io
 import base64
 from flask import Flask, request, jsonify
@@ -269,9 +269,25 @@ from flask_cors import CORS
 app = Flask(__name__)
 CORS(app)
 
+# Function to check disk usage
+def check_disk_space():
+    result = subprocess.run(['df', '-h'], capture_output=True, text=True)
+    print(result.stdout)
+
+# Function to clear Hugging Face cache
+def clear_huggingface_cache():
+    cache_dir = os.path.expanduser('~/.cache/huggingface')
+    if os.path.exists(cache_dir):
+        shutil.rmtree(cache_dir)  # Removes the entire cache directory
+        print(f"Cleared Hugging Face cache at: {cache_dir}")
+    else:
+        print("No Hugging Face cache found.")
+
+# Check disk space
+check_disk_space()
 
 # Clear Hugging Face cache
-HfFolder.delete_cache()
+clear_huggingface_cache()
 
 # Add config to store base64 images
 app.config['image_outputs'] = {}