Update app.py

app.py

@@ -8,11 +8,9 @@ from gradio import Chatbot
 from streaming import stream_to_gradio
 from huggingface_hub import login
 from gradio.data_classes import FileData
-
-
-
 import torch
 
+
 # Initialize ZeroGPU
 if torch.cuda.is_available():
     torch.backends.cudnn.benchmark = True
@@ -29,6 +27,9 @@ if torch.cuda.is_available():
     ZeroGpu(use_cpu=False)
 
 
+# Set HF_HOME
+os.environ['HF_HOME'] = 'C:/Users/jpkor/data/.huggingface'
+
 login(os.getenv("HUGGINGFACEHUB_API_TOKEN"), add_to_git_credential=True)