Spaces:

tdnathmlenthusiast
/

OCR_DEMO_TEST

Sleeping

tdnathmlenthusiast commited on Sep 29, 2024

Commit

23f773c

verified ·

1 Parent(s): cd6541c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,7 @@ import os
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 print(device)
 # load_desirable_model
-got_model_name = "tdnathmlenthusiast/td_ocr_2.0_hindi"
 det_processor, det_model = load_det_processor(), load_det_model()
 det_model.to(device)
 rec_model, rec_processor = load_rec_model(), load_rec_processor()

 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 print(device)
 # load_desirable_model
+got_model_name = "tdnathmlenthusiast/tester"
 det_processor, det_model = load_det_processor(), load_det_model()
 det_model.to(device)
 rec_model, rec_processor = load_rec_model(), load_rec_processor()