Update app.py

app.py

@@ -341,7 +341,7 @@ def load_models_and_documents():
             trust_remote_code=True,
             enforce_eager=True,
             quantization="awq",
-            gpu_memory_utilization=0.7,
+            gpu_memory_utilization=0.8,
             max_model_len=12288,
             dtype=torch.float16,
             max_num_seqs=128
@@ -421,7 +421,7 @@ def load_models_and_documents():
 
             st.write('Downloading and Loading Video Games Dataset coming from Wikipedia...')
 
-            dataset = pd.read_parquet(os.path.join(os.getenv('HF_HOME'), 'train_pages.parquet.zst'), engine='pyarrow')
+            dataset = pd.read_parquet(os.path.join(os.getenv('HOME'),'data', 'train_pages.parquet.zst'), engine='pyarrow')
 
             for columnName, columnData in dataset.iteritems():
                 if columnName == 'text':