resume download

app.py

@@ -13,7 +13,7 @@ from typing import Any
 
 
 from huggingface_hub import hf_hub_download
-ckpt_path = hf_hub_download(repo_id="Viglong/OriNet", filename="celarge/dino_weight.pt", repo_type="model", cache_dir='./')
+ckpt_path = hf_hub_download(repo_id="Viglong/OriNet", filename="celarge/dino_weight.pt", repo_type="model", cache_dir='./', resume_download=True)
 print(ckpt_path)
 
 save_path = './'
@@ -146,12 +146,13 @@ def figure_to_img(fig):
     return image
 
 def infer_func(img, do_rm_bkg):
+    img = Image.fromarray(img)
     img = background_preprocess(img, do_rm_bkg)
     angles = get_3angle(img)
     
     fig, ax = plt.subplots(figsize=(8, 8))
 
-    h, w, c = img.shape
+    w, h = img.size
     if h>w:
         extent = [-5*w/h, 5*w/h, -5, 5]
     else: