Hugging Face's logo

Spaces:
vishnun
/
CLIPnCROP
Running

App Files Community
vishnun commited on
Commit
ae343dc
·
1 Parent(s): a00839f

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, YolosImageProcessor, YolosForObjectDetection
5
  import torch
6
 
7
- feature_extractor = YolosImageProcessor.from_pretrained("hustvl/yolos-small")
8
- dmodel = YolosForObjectDetection.from_pretrained('hustvl/yolos-small')
9
 
10
  model = CLIPModel.from_pretrained("openai/clip-vit-base-patch16")
11
  processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch16")
 
4
  from transformers import CLIPProcessor, CLIPModel, YolosImageProcessor, YolosForObjectDetection
5
  import torch
6
 
7
+ feature_extractor = YolosImageProcessor.from_pretrained("hustvl/yolos-tiny")
8
+ dmodel = YolosForObjectDetection.from_pretrained('hustvl/yolos-tiny')
9
 
10
  model = CLIPModel.from_pretrained("openai/clip-vit-base-patch16")
11
  processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch16")