DravenBlack commited on
Commit
34e935f
·
verified ·
1 Parent(s): a712336

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -1
app.py CHANGED
@@ -6,7 +6,13 @@ import torch
6
  from PIL import Image
7
  import io
8
  import sys
9
- sys.path.append("./OmniParser") # ✨加入这一行,修正路径
 
 
 
 
 
 
10
 
11
  import base64, os
12
  from util.utils import check_ocr_box, get_yolo_model, get_caption_model_processor, get_som_labeled_img
 
6
  from PIL import Image
7
  import io
8
  import sys
9
+ import os
10
+
11
+ # 把OmniParser目录加入到Python路径
12
+ sys.path.append(os.path.join(os.path.dirname(__file__), "OmniParser"))
13
+
14
+ # 引入gradio_demo里的demo
15
+ from gradio_demo import demo
16
 
17
  import base64, os
18
  from util.utils import check_ocr_box, get_yolo_model, get_caption_model_processor, get_som_labeled_img