Spaces:
Running
Running
Doron Adler
commited on
Commit
•
9bb320a
1
Parent(s):
d9f299b
Use MPS device (via Pytorch for Apple silicon) when available instead of cpu
Browse files
app.py
CHANGED
@@ -54,7 +54,7 @@ else:
|
|
54 |
processor = AutoProcessor.from_pretrained("microsoft/Florence-2-base-ft", trust_remote_code=True)
|
55 |
model.to(device)
|
56 |
|
57 |
-
DESCRIPTION = "# [Florence-2 base-ft Demo with CPU inference support](https://huggingface.co/microsoft/Florence-2-base-ft)"
|
58 |
|
59 |
colormap = ['blue','orange','green','purple','brown','pink','gray','olive','cyan','red',
|
60 |
'lime','indigo','violet','aqua','magenta','coral','gold','tan','skyblue']
|
|
|
54 |
processor = AutoProcessor.from_pretrained("microsoft/Florence-2-base-ft", trust_remote_code=True)
|
55 |
model.to(device)
|
56 |
|
57 |
+
DESCRIPTION = "# [Florence-2 base-ft Demo with CPU and MPS inference support](https://huggingface.co/microsoft/Florence-2-base-ft)"
|
58 |
|
59 |
colormap = ['blue','orange','green','purple','brown','pink','gray','olive','cyan','red',
|
60 |
'lime','indigo','violet','aqua','magenta','coral','gold','tan','skyblue']
|