chriswang09 commited on
Commit
3acc03a
Β·
1 Parent(s): a7b748b
Files changed (3) hide show
  1. README.md +2 -1
  2. app.py +6 -0
  3. packages.txt +5 -0
README.md CHANGED
@@ -3,7 +3,8 @@ title: Ui Convert2
3
  emoji: πŸ†
4
  colorFrom: indigo
5
  colorTo: red
6
- sdk: docker
 
7
  app_file: app.py
8
  pinned: false
9
  ---
 
3
  emoji: πŸ†
4
  colorFrom: indigo
5
  colorTo: red
6
+ sdk: gradio
7
+ sdk_version: 5.21.0
8
  app_file: app.py
9
  pinned: false
10
  ---
app.py CHANGED
@@ -5,6 +5,12 @@ import numpy as np
5
  import json
6
  from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
7
  import pytesseract
 
 
 
 
 
 
8
 
9
  # Load Object Detection Pipeline
10
  obj_detect = pipeline("object-detection", model="facebook/detr-resnet-50", device=-1)
 
5
  import json
6
  from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
7
  import pytesseract
8
+ import os
9
+
10
+ # Set cache directories to /tmp which should be writable
11
+ os.environ["TRANSFORMERS_CACHE"] = "/tmp/transformers_cache"
12
+ os.environ["HF_HOME"] = "/tmp/hf_home"
13
+ os.environ["HF_DATASETS_CACHE"] = "/tmp/hf_datasets_cache"
14
 
15
  # Load Object Detection Pipeline
16
  obj_detect = pipeline("object-detection", model="facebook/detr-resnet-50", device=-1)
packages.txt CHANGED
@@ -1,2 +1,7 @@
1
  tesseract-ocr
2
  libtesseract-dev
 
 
 
 
 
 
1
  tesseract-ocr
2
  libtesseract-dev
3
+ libgl1-mesa-glx
4
+ libglib2.0-0
5
+ libsm6
6
+ libxrender1
7
+ libxext6