GnanaPrasath commited on
Commit
bef264e
·
verified ·
1 Parent(s): 8770457

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -1,5 +1,8 @@
1
  import gradio as gr
2
  from ocr_tamil.ocr import OCR
 
 
 
3
 
4
  model_path = r"parseq_tamil_v6.ckpt"
5
  text_detect_model = r"craft_mlt_25k.pth" # add the full path of the craft model
 
1
  import gradio as gr
2
  from ocr_tamil.ocr import OCR
3
+ import torch
4
+
5
+ torch.hub.load('baudm/parseq', name, pretrained=True)
6
 
7
  model_path = r"parseq_tamil_v6.ckpt"
8
  text_detect_model = r"craft_mlt_25k.pth" # add the full path of the craft model