mrfakename commited on
Commit
142c37c
1 Parent(s): 4db061f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ model_id = "yifeihu/TB-OCR-preview-0.1"
21
  DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
22
 
23
  if not torch.cuda.is_available():
24
- ABOUT += "\n\n### This demo is running on CPU, it will be very slow..."
25
 
26
  model = AutoModelForCausalLM.from_pretrained(
27
  model_id,
 
21
  DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
22
 
23
  if not torch.cuda.is_available():
24
+ ABOUT += "\n\n### This demo is running on CPU\n\nThis demo is running on CPU, it will be very slow. Consider duplicating it or running it locally to skip the queue and for faster response times."
25
 
26
  model = AutoModelForCausalLM.from_pretrained(
27
  model_id,