Update run.py

run.py

@@ -8,23 +8,31 @@ import re
 from werkzeug.utils import secure_filename
 import uuid
 
+import os
+import platform
+
+# Set Transformers Cache Directory
 if platform.system() == "Windows":
-    print("Windows detected. Assigning cache directory to Transformers in AppData\Local.")
+    print("Windows detected. Assigning cache directory to Transformers in AppData\\Local.")
     transformers_cache_directory = os.path.join(os.getenv('LOCALAPPDATA'), 'transformers_cache')
-    if not os.path.exists(transformers_cache_directory):
-        try:
-            os.mkdir(transformers_cache_directory)
-            print(f"First launch. Directory '{transformers_cache_directory}' created successfully.")
-        except OSError as e:
-            print(f"Error creating directory '{transformers_cache_directory}': {e}")
-    else:
-        print(f"Directory '{transformers_cache_directory}' already exists.")
-    os.environ['TRANSFORMERS_CACHE'] = transformers_cache_directory
-    print("Environment variable assigned.")
-    del transformers_cache_directory
+else:
+    print("Non-Windows system detected. Assigning cache directory to /tmp/transformers_cache.")
+    transformers_cache_directory = '/tmp/transformers_cache'
 
+# Ensure the directory exists
+if not os.path.exists(transformers_cache_directory):
+    try:
+        os.makedirs(transformers_cache_directory, exist_ok=True)
+        print(f"Directory '{transformers_cache_directory}' created successfully.")
+    except OSError as e:
+        print(f"Error creating directory '{transformers_cache_directory}': {e}")
 else:
-    print("Windows not detected. Assignment of Transformers cache directory not necessary.")
+    print(f"Directory '{transformers_cache_directory}' already exists.")
+
+# Set the TRANSFORMERS_CACHE environment variable
+os.environ['TRANSFORMERS_CACHE'] = transformers_cache_directory
+print(f"Environment variable TRANSFORMERS_CACHE set to '{transformers_cache_directory}'.")
+
 
 class Config:
     UPLOAD_FOLDER = os.path.join(os.path.dirname(__file__), 'uploads')  # Correct path