Spaces:
Running
on
Zero
Running
on
Zero
mrfakename
commited on
Commit
•
142c37c
1
Parent(s):
4db061f
Update app.py
Browse files
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,
|