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
|