Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -46,7 +46,7 @@ try:
|
|
46 |
except NotImplementedError:
|
47 |
from transformers import pipeline
|
48 |
|
49 |
-
pipe = pipeline("image-classification",
|
50 |
|
51 |
|
52 |
def face_detector(input_img):
|
|
|
46 |
except NotImplementedError:
|
47 |
from transformers import pipeline
|
48 |
|
49 |
+
pipe = pipeline("image-classification", model, device=device)
|
50 |
|
51 |
|
52 |
def face_detector(input_img):
|