Update app.py

app.py

@@ -142,7 +142,7 @@ def skintoneplot(hex_codes):
     return fig
 
 def age_detector(image):
-    pipe = pipeline('image-classification', model="dima806/faces_age_detection", device=0)
+    pipe = pipeline('image-classification', model="dima806/faces_age_detection", device="cuda")
     result = pipe(image)
     max_score_item = max(result, key=lambda item: item['score'])
     return max_score_item['label']
@@ -160,7 +160,7 @@ def ageplot(agelist):
     return fig
 
 def is_nsfw(image):
-    classifier = pipeline("image-classification", model="Falconsai/nsfw_image_detection")
+    classifier = pipeline("image-classification", model="Falconsai/nsfw_image_detection", device="cuda")
     result = classifier(image)
     max_score_item = max(result, key=lambda item: item['score'])
     return max_score_item['label']