merve HF staff commited on
Commit
7d7b08f
·
verified ·
1 Parent(s): 987cfe1

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -0
app.py ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from transformers import AutoModel, CLIPImageProcessor, CLIPTokenizer
2
+ import torch
3
+
4
+ model_name_or_path = "BAAI/EVA-CLIP-8B"
5
+ image_size = 224
6
+
7
+ processor = CLIPImageProcessor.from_pretrained("openai/clip-vit-large-patch14")
8
+
9
+ model = AutoModel.from_pretrained(
10
+ model_name_or_path,
11
+ torch_dtype=torch.bfloat16,
12
+ trust_remote_code=True).to('cuda').eval()
13
+
14
+
15
+ tokenizer = CLIPTokenizer.from_pretrained(model_name_or_path)