SoyVitou commited on
Commit
aefb995
·
verified ·
1 Parent(s): 4702123

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -3,7 +3,7 @@ import importlib.util
3
  from huggingface_hub import hf_hub_download
4
  import gradio as gr
5
 
6
- PRIVATE_REPO_ID = "SoyVitou/infinity-khmer-ocr-large"
7
  infinity_model = "model.py"
8
 
9
  EXAMPLES = [
@@ -25,9 +25,9 @@ def import_private_module(repo_id: str, filename: str, token: str):
25
 
26
  HF_TOKEN = os.environ.get("HF_TOKEN", "").strip()
27
  if not HF_TOKEN:
28
- raise RuntimeError("HF_TOKEN not found. Add it in Space Settings → Secrets as HF_TOKEN.")
29
 
30
- pm = import_private_module(PRIVATE_REPO_ID, infinity_model, HF_TOKEN)
31
  MODEL, TOK, CFG_OBJ, DEVICE = pm.load_from_private_repo(hf_hub_download, HF_TOKEN)
32
 
33
 
@@ -39,7 +39,7 @@ demo = gr.Interface(
39
  fn=predict,
40
  inputs=gr.Image(type="pil", label="Upload image"),
41
  outputs=gr.Textbox(label="OCR result", lines=6),
42
- title="Infinity Khmer OCR",
43
  examples=EXAMPLES,
44
  cache_examples=False,
45
  )
 
3
  from huggingface_hub import hf_hub_download
4
  import gradio as gr
5
 
6
+ infinity_repo_2026 = "SoyVitou/infinity-khmer-ocr-large"
7
  infinity_model = "model.py"
8
 
9
  EXAMPLES = [
 
25
 
26
  HF_TOKEN = os.environ.get("HF_TOKEN", "").strip()
27
  if not HF_TOKEN:
28
+ raise RuntimeError("HF_TOKEN not found.")
29
 
30
+ pm = import_private_module(infinity_repo_2026, infinity_model, HF_TOKEN)
31
  MODEL, TOK, CFG_OBJ, DEVICE = pm.load_from_private_repo(hf_hub_download, HF_TOKEN)
32
 
33
 
 
39
  fn=predict,
40
  inputs=gr.Image(type="pil", label="Upload image"),
41
  outputs=gr.Textbox(label="OCR result", lines=6),
42
+ title="Infinity Large Khmer OCR",
43
  examples=EXAMPLES,
44
  cache_examples=False,
45
  )