fix detectron2 installation

README.md

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

app.py

@@ -1,3 +1,10 @@
+import os
+
+# build detectron2 from source
+# we can't build detectron2 in requirements.txt because it needs PyTorch installed first,
+# but requirements.txt will try to build wheels before installing any packages.
+os.system("pip install git+https://github.com/facebookresearch/detectron2.git")
+
 import gradio as gr
 import numpy as np
 from transformers import LayoutLMv2Processor, LayoutLMv2ForTokenClassification

requirements.txt

@@ -1,10 +1,8 @@
 --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
-torch==1.10.1+cpu
-torchvision==0.11.2+cpu
+torch
+torchvision
 transformers
-detectron2
 pytesseract