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
Files changed (1) hide show
  1. app.py +1 -1
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']