Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
b2c8d9d
1
Parent(s):
1c9ce50
Fix the JIT related error (#10)
Browse files- Update app.py (abc1a713812a9b5de18026c33b8b27f0f023cf68)
Co-authored-by: hysts <hysts@users.noreply.huggingface.co>
app.py
CHANGED
|
@@ -5,6 +5,10 @@ import numpy as np
|
|
| 5 |
import time
|
| 6 |
import random
|
| 7 |
from PIL import Image
|
|
|
|
|
|
|
|
|
|
|
|
|
| 8 |
from transparent_background import Remover
|
| 9 |
|
| 10 |
@spaces.GPU()
|
|
|
|
| 5 |
import time
|
| 6 |
import random
|
| 7 |
from PIL import Image
|
| 8 |
+
import torch
|
| 9 |
+
|
| 10 |
+
torch.jit.script = lambda f: f
|
| 11 |
+
|
| 12 |
from transparent_background import Remover
|
| 13 |
|
| 14 |
@spaces.GPU()
|