Update app/run.py

app/run.py

@@ -1,13 +1,10 @@
 import os
 
-# Create the cache directory if it doesn't exist and set permissions
-hf_cache_dir = '/app/hf_cache'
-os.makedirs(hf_cache_dir, exist_ok=True)
-
-# Set the HF_HOME environment variable
-os.environ['HF_HOME'] = hf_cache_dir
+# Ensure the HF_HOME environment variable is set correctly
+hf_cache_dir = os.getenv('HF_HOME', '/app/hf_cache')
 
-# Application code starts here
+# Ensure the directory exists
+os.makedirs(hf_cache_dir, exist_ok=True)
 
 import logging
 from flask import Flask, request, jsonify