qiqiyuan commited on
Commit
a20e699
·
verified ·
1 Parent(s): 4de53df

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
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
- #model_path = "best_int8_openvino_model"
 
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 = "ITI107-2024S2-4759245U"
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,