2023-11-27-07-25-56

app.py

@@ -1,13 +1,15 @@
 
 import os
 
-os.system("pip install torch")
-os.system("pip install 'git+https://github.com/facebookresearch/detectron2.git'")
-os.system("pip install layoutparser")
-os.system("pip install layoutparser[layoutmodels]")
-os.system("pip install layoutparser[ocr]")
+os.system("python3 -m venv mini-env")
+os.system("source mini-env/bin/activate")
+
+os.system("pip install torch==2.1.0+cu118 torchvision torchaudio")
+os.system("pip install 'git+https://github.com/facebookresearch/detectron2.git@v0.6'")
+os.system("pip install layoutparser==0.3.4 layoutparser[layoutmodels] layoutparser[ocr]")
 os.system("pip install Pillow==9.4.0")
-os.system("pip install requests")
+os.system("pip install requests==2.31.0")
+os.system("pip install -q gradio==4.7.1")
 
 import gradio as gr
 import layoutparser as lp
@@ -78,7 +80,7 @@ def inference_factory(config_path: str, model_path: str, label_map: dict, color_
             return error if just_image else [None, error, threshold]
     if not launch:
         return fn
-    
+
     ###########################################################
     ################### Start of Gradio setup #################
     ###########################################################