update app.py

app.py

@@ -5,8 +5,7 @@ warnings.filterwarnings('ignore')
 import subprocess, io, os, sys, time
 # os.system("pip install gradio==3.36.1")
 os.system("pip install gradio==3.40.1")
-# os.system("pip install transformers==4.33.2")
-# os.system("pip install transformers==4.30.0")
+# os.system("pip install transformers==4.32.0")
 import gradio as gr
 
 from loguru import logger
@@ -76,6 +75,9 @@ if lama_cleaner_enable:
 import ast
 from kosmos_utils import *
 
+ram_enable = True
+kosmos_enable = True
+
 config_file = 'GroundingDINO/groundingdino/config/GroundingDINO_SwinT_OGC.py'
 ckpt_repo_id = "ShilongLiu/GroundingDINO"
 ckpt_filenmae = "groundingdino_swint_ogc.pth"
@@ -794,9 +796,6 @@ def get_model_device(module):
     except Exception as e:
         return 'Error'
 
-ram_enable = True
-kosmos_enable = False
-
 task_types = ["detection", "segment", "inpainting"]
 if lama_cleaner_enable:
     task_types.append("remove")
@@ -812,6 +811,9 @@ if __name__ == "__main__":
     args = parser.parse_args()
     print(f'args = {args}')
 
+    if os.environ.get('IS_MY_DEBUG') is None:
+        os.system("pip list")
+    
     set_device()
 
     if os.environ.get('IS_MY_DEBUG') is None and kosmos_enable:

requirements.txt

@@ -17,8 +17,6 @@ termcolor
 timm
 torch
 torchvision
-transformers==4.27.4
-# transformers@git+https://github.com/huggingface/transformers.git@main
 yapf
 numba
 scipy 
@@ -28,5 +26,9 @@ sentencepiece
 loguru
 
 lama-cleaner==1.2.3
+
+transformers==4.32.0
+# transformers@git+https://github.com/huggingface/transformers.git@main
+
 openmim==0.3.9
 mmcv==2.0.0