fix deps

README.md

@@ -6,6 +6,7 @@ colorTo: green
 sdk: gradio
 app_file: app.py
 pinned: false
+python_version: "3.9"
 ---
 
 # Configuration

app.py

@@ -1,9 +1,3 @@
-import os
-os.system('pip install pyyaml==5.1')
-os.system('pip install torch==1.10.1+cpu torchvision==0.11.2+cpu -f https://download.pytorch.org/whl/cpu')
-os.system('pip install git+https://github.com/facebookresearch/detectron2.git')
-os.system('pip install pytesseract')
-
 import gradio as gr
 import numpy as np
 from transformers import LayoutLMv2Processor, LayoutLMv2ForTokenClassification

requirements.txt

@@ -1,6 +1,10 @@
-gradio
+--index-url https://download.pytorch.org/whl/cpu
+--extra-index-url https://dl.fbaipublicfiles.com/detectron2/wheels/cpu/torch1.10/index.html
+--extra-index-url https://pypi.org/simple
 Pillow
 numpy
-datasets
-torch
-transformers
+torch==1.10.1+cpu
+torchvision==0.11.2+cpu
+transformers
+detectron2
+pytesseract