Spaces:

tdnathmlenthusiast
/

OCR-for-Hindi-English

Running

tdnathmlenthusiast commited on Sep 29

Commit

1543d7d

•

1 Parent(s): 8cafc94

fixing model loading issues

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/got-ocr-2.0-modified-hindi-version'
 det_processor, det_model = load_det_processor(), load_det_model()
 det_model.to(device)

 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 print(device)
 # load_desirable_model
+got_model_name = 'aarishshahmohsin/got_ocr_2'
 det_processor, det_model = load_det_processor(), load_det_model()
 det_model.to(device)