chaishuaishuai commited on
Commit
5f10a77
1 Parent(s): cf0ec70

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +23 -0
app.py ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # test.py
2
+ import torch
3
+ from PIL import Image
4
+ from transformers import AutoModel, AutoTokenizer
5
+
6
+ model = AutoModel.from_pretrained('openbmb/MiniCPM-Llama3-V-2_5', trust_remote_code=True, torch_dtype=torch.float16)
7
+ model = model.to(device='cuda')
8
+
9
+ tokenizer = AutoTokenizer.from_pretrained('openbmb/MiniCPM-Llama3-V-2_5', trust_remote_code=True)
10
+ model.eval()
11
+
12
+ image = Image.open('xx.jpg').convert('RGB')
13
+ question = 'What is in the image?'
14
+ msgs = [{'role': 'user', 'content': question}]
15
+
16
+ res = model.chat(
17
+ image=image,
18
+ msgs=msgs,
19
+ tokenizer=tokenizer,
20
+ sampling=True,
21
+ temperature=0.7
22
+ )
23
+ print(res)