Inference Endpoints
Vishakaraj commited on
Commit
72a2e39
1 Parent(s): 72e9733

Debug handler

Browse files
Files changed (1) hide show
  1. handler.py +5 -5
handler.py CHANGED
@@ -1,6 +1,6 @@
1
  import os
2
 
3
- os.system("ls /app")
4
  os.system("pip install deepspeed==0.7.0")
5
 
6
  from PIL import Image
@@ -10,10 +10,10 @@ import sys
10
  import numpy as np
11
  import torch
12
 
13
- from detectron2.config import get_cfg
14
- from detectron2.data.detection_utils import read_image
15
- from detectron2.utils.logger import setup_logger
16
-
17
  sys.path.insert(0, "third_party/CenterNet2/projects/CenterNet2/")
18
  from centernet.config import add_centernet_config
19
  from grit.config import add_grit_config
 
1
  import os
2
 
3
+ os.system("pip install detectron2-0.6-cp39-cp39-linux_x86_64.whl")
4
  os.system("pip install deepspeed==0.7.0")
5
 
6
  from PIL import Image
 
10
  import numpy as np
11
  import torch
12
 
13
+ # from detectron2.config import get_cfg
14
+ # from detectron2.data.detection_utils import read_image
15
+ # from detectron2.utils.logger import setup_logger
16
+ print("###################################", os.getcwd())
17
  sys.path.insert(0, "third_party/CenterNet2/projects/CenterNet2/")
18
  from centernet.config import add_centernet_config
19
  from grit.config import add_grit_config