asrs777 commited on
Commit
76ce779
1 Parent(s): 47e7826

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ device = 1 if torch.cuda.is_available() else "cpu"
12
 
13
  # chk_point = "kdhht2334/autotrain-diffusion-emotion-facial-expression-recognition-40429105176"
14
 
15
- model = AutoModelForImageClassification.from_pretrained(chk_point)
16
 
17
  ##Add face detector
18
  from facenet_pytorch import MTCNN, InceptionResnetV1
 
12
 
13
  # chk_point = "kdhht2334/autotrain-diffusion-emotion-facial-expression-recognition-40429105176"
14
 
15
+ model = AutoModelForImageClassification.from_pretrained("./autotrain-diffusion-emotion-facial-expression-recognition-40429105176")
16
 
17
  ##Add face detector
18
  from facenet_pytorch import MTCNN, InceptionResnetV1