Update app.py

app.py

@@ -1,11 +1,5 @@
 import os
 
-os.environ["CUDA_VISIBLE_DEVICES"] = "1,6" # to use the GPUs 3,4 only
-
-os.environ["HF_HUB_CACHE"] = "/eos/jeodpp/home/users/consose/cache/huggingface/hub"
-os.environ["HUGGINGFACE_HUB_CACHE"] = "/eos/jeodpp/home/users/consose/cache/huggingface/hub"
-os.environ["HF_HOME"] = "/eos/jeodpp/home/users/consose/cache/huggingface/hub"
-
 from transformers import file_utils
 print(file_utils.default_cache_path)
 
@@ -27,10 +21,10 @@ from concurrent.futures import ThreadPoolExecutor, as_completed
 from collections import Counter
 
 #os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "max_split_size_mb:512"
-os.environ['PYTORCH_CUDA_ALLOC_CONF'] = 'expandable_segments:True'
+#os.environ['PYTORCH_CUDA_ALLOC_CONF'] = 'expandable_segments:True'
 
 import torch
-torch.cuda.empty_cache()  # Clear cache ot torch
+#torch.cuda.empty_cache()  # Clear cache ot torch
 
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 print(f"Device: {device}...")