Update app.py

app.py

@@ -10,7 +10,8 @@ from langchain_chroma import Chroma
 from chromadb import Documents, EmbeddingFunction, Embeddings  
 from chromadb.config import Settings  
 import chromadb #import HttpClient 
-import os  
+import os
+import tempfile
 import re 
 import uuid  
 import gradio as gr  
@@ -27,6 +28,13 @@ os.environ['CUDA_LAUNCH_BLOCKING'] = '1'
 os.environ['CUDA_CACHE_DISABLE'] = '1'  
   
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")  
+
+# Ensure the temporary directory exists
+temp_dir = '/tmp/gradio/'
+os.makedirs(temp_dir, exist_ok=True)
+
+# Set Gradio cache directory
+gr.components.file.GRADIO_CACHE = temp_dir
   
 ### Utils  
 hf_token, yi_token = load_env_variables()