Spaces:
Sleeping
Sleeping
Update app.py with improved Gradio interface and change model's torch_dtype to float32
Browse files
app.py
CHANGED
@@ -3,10 +3,10 @@ import torch
|
|
3 |
from PIL import Image
|
4 |
from transformers import AutoModel, AutoTokenizer
|
5 |
|
6 |
-
device = "
|
7 |
|
8 |
# Load the model
|
9 |
-
model = AutoModel.from_pretrained("openbmb/MiniCPM-Llama3-V-2_5", trust_remote_code=True, torch_dtype=torch.
|
10 |
model = model.to(device=device)
|
11 |
|
12 |
# Load the tokenizer
|
|
|
3 |
from PIL import Image
|
4 |
from transformers import AutoModel, AutoTokenizer
|
5 |
|
6 |
+
device = "cpu"
|
7 |
|
8 |
# Load the model
|
9 |
+
model = AutoModel.from_pretrained("openbmb/MiniCPM-Llama3-V-2_5", trust_remote_code=True, torch_dtype=torch.float32)
|
10 |
model = model.to(device=device)
|
11 |
|
12 |
# Load the tokenizer
|