vincentclaes commited on
Commit
262aca1
1 Parent(s): fd2ef9c

use auth token

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -10,10 +10,12 @@ APP_NAME = "Mona Lisa Detection"
10
 
11
  logger.debug("loading processor and model.")
12
  processor = AutoFeatureExtractor.from_pretrained(
13
- "drift-ai/autotrain-mona-lisa-detection-38345101350"
 
14
  )
15
  model = AutoModelForImageClassification.from_pretrained(
16
- "drift-ai/autotrain-mona-lisa-detection-38345101350"
 
17
  )
18
  logger.debug("loading processor and model succeeded.")
19
 
 
10
 
11
  logger.debug("loading processor and model.")
12
  processor = AutoFeatureExtractor.from_pretrained(
13
+ "drift-ai/autotrain-mona-lisa-detection-38345101350",
14
+ use_auth_token=True
15
  )
16
  model = AutoModelForImageClassification.from_pretrained(
17
+ "drift-ai/autotrain-mona-lisa-detection-38345101350",
18
+ use_auth_token=True
19
  )
20
  logger.debug("loading processor and model succeeded.")
21