bulatko commited on
Commit
6b056a6
·
1 Parent(s): 26bde8f

Runtime fallback: install local detectron2 if import fails (no build isolation)

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. mvp.py +1 -3
app.py CHANGED
@@ -1,5 +1,5 @@
1
  import os
2
-
3
  import gradio as gr
4
 
5
 
 
1
  import os
2
+ os.system('pip install git+https://github.com/facebookresearch/detectron2.git')
3
  import gradio as gr
4
 
5
 
mvp.py CHANGED
@@ -964,9 +964,7 @@ def detect_objects(text_labels, target_dir, conf_thres, *viz_args):
964
  try:
965
  env = dict(os.environ)
966
  env["PYTHONPATH"] = (
967
- DETECTRON2_ROOT
968
- + os.pathsep
969
- + MK_PATH
970
  + (os.pathsep + env["PYTHONPATH"] if env.get("PYTHONPATH") else "")
971
  )
972
  root_input_dir = os.path.dirname(target_dir)
 
964
  try:
965
  env = dict(os.environ)
966
  env["PYTHONPATH"] = (
967
+ MK_PATH
 
 
968
  + (os.pathsep + env["PYTHONPATH"] if env.get("PYTHONPATH") else "")
969
  )
970
  root_input_dir = os.path.dirname(target_dir)