vishnun commited on
Commit
c2e7eaf
·
1 Parent(s): d858364

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -4,8 +4,8 @@ from PIL import Image
4
  from transformers import CLIPProcessor, CLIPModel, DetrFeatureExtractor, DetrForObjectDetection, AutoFeatureExtractor, AutoModelForObjectDetection
5
  import torch
6
 
7
- feature_extractor = AutoFeatureExtractor.from_pretrained("nielsr/detr-resnet-50")
8
- dmodel = AutoModelForObjectDetection.from_pretrained("nielsr/detr-resnet-50")
9
 
10
  model = CLIPModel.from_pretrained("openai/clip-vit-base-patch32")
11
  processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch32")
 
4
  from transformers import CLIPProcessor, CLIPModel, DetrFeatureExtractor, DetrForObjectDetection, AutoFeatureExtractor, AutoModelForObjectDetection
5
  import torch
6
 
7
+ feature_extractor = AutoFeatureExtractor.from_pretrained("facebook/detr-resnet-50")
8
+ dmodel = AutoModelForObjectDetection.from_pretrained("facebook/detr-resnet-50")
9
 
10
  model = CLIPModel.from_pretrained("openai/clip-vit-base-patch32")
11
  processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch32")