feat: init lama in gpu env

app.py

@@ -5,15 +5,15 @@ import torch
 from PIL import Image
 import numpy as np
 import cv2
+import os
 from simple_lama_inpainting import SimpleLama
 
-zero = torch.Tensor([0]).cuda()
-print(zero.device)  # <-- 'cpu' 🤔
-lama: SimpleLama = SimpleLama(device=zero.device)
-
+big_lama_url = os.path.join(os.path.abspath(os.path.dirname(__file__)), 'model/big-lama.pt')
 
 @spaces.GPU
 def lama_inpainting(image: np.ndarray, mask: np.ndarray) -> np.ndarray:
+    os.environ['LAMA_MODEL'] = big_lama_url
+    lama: SimpleLama = SimpleLama()
     res = lama(
         Image.fromarray(cv2.cvtColor(image, cv2.COLOR_BGR2RGB)).convert("RGB"),
         Image.fromarray(cv2.cvtColor(mask, cv2.COLOR_BGR2RGB)).convert("L"),

model/big-lama.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7ba7aa7ac37a4d41fdbbeba3a2af7ead18058552997e3a3cd1a3b2210c9e6b4c
+size 205803670