Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -3,12 +3,11 @@ from PIL import Image
|
|
3 |
import gradio as gr
|
4 |
from huggingface_hub import snapshot_download
|
5 |
import os
|
6 |
-
import patoolib
|
7 |
|
8 |
-
|
|
|
9 |
|
10 |
def load_model(repo_id):
|
11 |
-
patoolib.extract_archive("best_int8_openvino_model.zip")
|
12 |
download_dir = snapshot_download(repo_id)
|
13 |
print(download_dir)
|
14 |
path = os.path.join(download_dir, "best_int8_openvino_model")
|
@@ -29,7 +28,7 @@ def predict(pilimg):
|
|
29 |
return out_pilimg
|
30 |
|
31 |
|
32 |
-
REPO_ID = "
|
33 |
detection_model = load_model(REPO_ID)
|
34 |
|
35 |
gr.Interface(fn=predict,
|
|
|
3 |
import gradio as gr
|
4 |
from huggingface_hub import snapshot_download
|
5 |
import os
|
|
|
6 |
|
7 |
+
|
8 |
+
model_path = "best_int8_openvino_model"
|
9 |
|
10 |
def load_model(repo_id):
|
|
|
11 |
download_dir = snapshot_download(repo_id)
|
12 |
print(download_dir)
|
13 |
path = os.path.join(download_dir, "best_int8_openvino_model")
|
|
|
28 |
return out_pilimg
|
29 |
|
30 |
|
31 |
+
REPO_ID = "qiqiyuan/4759245U"
|
32 |
detection_model = load_model(REPO_ID)
|
33 |
|
34 |
gr.Interface(fn=predict,
|