Commit
·
5b3ddda
1
Parent(s):
36de9cf
Remove gradio import
Browse files- handler.py +2 -3
handler.py
CHANGED
@@ -1,7 +1,7 @@
|
|
1 |
import os
|
2 |
|
3 |
-
|
4 |
-
|
5 |
|
6 |
import site
|
7 |
from importlib import reload
|
@@ -14,7 +14,6 @@ import argparse
|
|
14 |
import sys
|
15 |
import numpy as np
|
16 |
import torch
|
17 |
-
import gradio as gr
|
18 |
|
19 |
from detectron2.config import get_cfg
|
20 |
from detectron2.data.detection_utils import read_image
|
|
|
1 |
import os
|
2 |
|
3 |
+
os.system("cd detectron2 && pip install detectron2-0.6-cp310-cp310-linux_x86_64.whl")
|
4 |
+
os.system("pip install deepspeed==0.7.0")
|
5 |
|
6 |
import site
|
7 |
from importlib import reload
|
|
|
14 |
import sys
|
15 |
import numpy as np
|
16 |
import torch
|
|
|
17 |
|
18 |
from detectron2.config import get_cfg
|
19 |
from detectron2.data.detection_utils import read_image
|