Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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(
|
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
|