Hugging Face's logo

Spaces:
vishnun
/
CLIPnCROP
Running

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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,8 +7,8 @@ import torch
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-patch32")
11
- processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch32")
12
 
13
  i1 = gr.Image(type="pil", label="Input image")
14
  i2 = gr.Textbox(label="Description for section to extracted")
 
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")
12
 
13
  i1 = gr.Image(type="pil", label="Input image")
14
  i2 = gr.Textbox(label="Description for section to extracted")