huylaughmad commited on
Commit
da34db4
·
verified ·
1 Parent(s): 075389d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -6
app.py CHANGED
@@ -1,23 +1,24 @@
 
 
 
 
 
1
  from flask import Flask, render_template
2
  from datasets import load_dataset
3
  from huggingface_hub import login
4
- import os
5
  import pandas as pd
6
 
7
  app = Flask(__name__)
8
 
9
- # Thiết lập thư mục cache cho Hugging Face
10
- os.environ["HF_HOME"] = "/app/hf_cache" # Đặt thư mục cache trong /app (có quyền ghi)
11
-
12
  # Đăng nhập Hugging Face
13
  HF_TOKEN = os.getenv("HF_TOKEN")
14
- print(f"HF_TOKEN: {HF_TOKEN}") # Debug token
15
  login(token=HF_TOKEN)
16
 
17
  # Tải dataset
18
  try:
19
  dataset = load_dataset("huylaughmad/web", data_files={"case_studies": "case_studies.csv"}, use_auth_token=True)
20
- print(f"Dataset loaded: {dataset}") # Debug dataset
21
  except Exception as e:
22
  print(f"Error loading dataset: {e}")
23
  dataset = {"case_studies": []}
 
1
+ import os
2
+
3
+ # Thiết lập HF_HOME trước khi import huggingface_hub hoặc datasets
4
+ os.environ["HF_HOME"] = "/app/hf_cache"
5
+
6
  from flask import Flask, render_template
7
  from datasets import load_dataset
8
  from huggingface_hub import login
 
9
  import pandas as pd
10
 
11
  app = Flask(__name__)
12
 
 
 
 
13
  # Đăng nhập Hugging Face
14
  HF_TOKEN = os.getenv("HF_TOKEN")
15
+ print(f"HF_TOKEN: {HF_TOKEN}")
16
  login(token=HF_TOKEN)
17
 
18
  # Tải dataset
19
  try:
20
  dataset = load_dataset("huylaughmad/web", data_files={"case_studies": "case_studies.csv"}, use_auth_token=True)
21
+ print(f"Dataset loaded: {dataset}")
22
  except Exception as e:
23
  print(f"Error loading dataset: {e}")
24
  dataset = {"case_studies": []}