mrfakename commited on
Commit
876e3b5
1 Parent(s): 142c37c

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\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,
 
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,