Spaces:

altayavci
/

OutfitChanger

Configuration error

altayavci commited on Jan 5

Commit

2d7bec9

•

1 Parent(s): 3500945

Update segmentation.py

Files changed (1) hide show

segmentation.py CHANGED Viewed

@@ -10,7 +10,7 @@ import cv2
 model = None
 extractor = None
-def init_body():
     global model, extractor
     extractor = AutoFeatureExtractor.from_pretrained("mattmdjaga/segformer_b2_clothes")
     model = SegformerForSemanticSegmentation.from_pretrained("mattmdjaga/segformer_b2_clothes").to("cuda")

 model = None
 extractor = None
+def init():
     global model, extractor
     extractor = AutoFeatureExtractor.from_pretrained("mattmdjaga/segformer_b2_clothes")
     model = SegformerForSemanticSegmentation.from_pretrained("mattmdjaga/segformer_b2_clothes").to("cuda")