Spaces:

ymo2017
/

ocr

Running on Zero

ymo2017 commited on Mar 13

Commit

47ffc6a

1 Parent(s): f7a1ff4

add more styling

Files changed (1) hide show

app.py CHANGED Viewed

@@ -20,8 +20,7 @@ from transformers.image_utils import load_image
 model_name = "stepfun-ai/GOT-OCR-2.0-hf"
-# device = "cuda" if torch.cuda.is_available() else "cpu"
-device = "cpu"
 processor = AutoProcessor.from_pretrained(model_name)
 model = AutoModelForImageTextToText.from_pretrained(

 model_name = "stepfun-ai/GOT-OCR-2.0-hf"
+device = "cuda" if torch.cuda.is_available() else "cpu"
 processor = AutoProcessor.from_pretrained(model_name)
 model = AutoModelForImageTextToText.from_pretrained(