using hf

face_parsing.py

@@ -7,14 +7,14 @@ from PIL import ImageOps
 import matplotlib.pyplot as plt
 
 def segmentation(input_img_path):
-    with open('face-parsing/config.json', 'r') as file:
-        data = json.load(file)
+    #with open('face-parsing/config.json', 'r') as file:
+     #   data = json.load(file)
 
-    for key, value in data["id2label"].items():
-        print(f"{key}: {value}")
+    #for key, value in data["id2label"].items():
+        #print(f"{key}: {value}")
 
-    image_processor = SegformerImageProcessor.from_pretrained("face-parsing")
-    model = SegformerForSemanticSegmentation.from_pretrained("face-parsing")
+    image_processor = SegformerImageProcessor.from_pretrained("jonathandinu/face-parsing")
+    model = SegformerForSemanticSegmentation.from_pretrained("jonathandinu/face-parsing")
 
     # input_img_path = "akshay kumar img.jpeg"
     if isinstance(input_img_path, str):  # It's a path