Update app/run.py

app/run.py

@@ -1,4 +1,15 @@
 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)
+os.chmod(hf_cache_dir, 0o777)
+
+# Set the HF_HOME environment variable
+os.environ['HF_HOME'] = hf_cache_dir
+
+# Application code starts here
+
 import logging
 from flask import Flask, request, jsonify
 from app.search_engine import PromptSearchEngine