Update models.py

models.py

@@ -3,12 +3,12 @@ from transformers import MaskFormerFeatureExtractor, MaskFormerForInstanceSegmen
 
 
 def get_detr():
-    image_processor = AutoImageProcessor.from_pretrained("facebook/detr-resnet-50-panoptic")
-    model = DetrForSegmentation.from_pretrained("facebook/detr-resnet-50-panoptic")
+    image_processor = AutoImageProcessor.from_pretrained("./processor/detr-resnet-50-panoptic")
+    model = DetrForSegmentation.from_pretrained("./models/detr-resnet-50-panoptic")
     return model, image_processor
 
 
 def get_maskformer():
-    feature_extractor = MaskFormerFeatureExtractor.from_pretrained("facebook/maskformer-swin-small-coco")
-    model = MaskFormerForInstanceSegmentation.from_pretrained("facebook/maskformer-swin-small-coco")
+    feature_extractor = MaskFormerFeatureExtractor.from_pretrained("./processor/maskformer-swin-small-coco")
+    model = MaskFormerForInstanceSegmentation.from_pretrained("./models/maskformer-swin-small-coco")
     return model, feature_extractor