ghengx commited on
Commit
80f0c7d
·
1 Parent(s): 70e05ba

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,7 +8,7 @@ warnings.filterwarnings("ignore")
8
  login(token=os.environ['HF_TOKEN'])
9
 
10
  repo = Repository(
11
- local_dir="",
12
  repo_type="dataset",
13
  clone_from=os.environ['DATASET'],
14
  token=True
@@ -16,7 +16,7 @@ repo = Repository(
16
  repo.git_pull()
17
 
18
  # Import the AI detector
19
- from model import AIContentDetector
20
 
21
  # Initialize the detector
22
  detector = AIContentDetector()
 
8
  login(token=os.environ['HF_TOKEN'])
9
 
10
  repo = Repository(
11
+ local_dir="dataset",
12
  repo_type="dataset",
13
  clone_from=os.environ['DATASET'],
14
  token=True
 
16
  repo.git_pull()
17
 
18
  # Import the AI detector
19
+ from dataset.model import AIContentDetector
20
 
21
  # Initialize the detector
22
  detector = AIContentDetector()