han1997 commited on
Commit
a83e4f5
·
verified ·
1 Parent(s): 274da13

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -1,5 +1,8 @@
1
  from __future__ import annotations
2
 
 
 
 
3
  import shlex
4
  import spaces
5
 
@@ -10,7 +13,6 @@ import hashlib
10
  import os
11
 
12
  from transformers import AutoModel, AutoProcessor
13
- import torch
14
  import sys
15
  import subprocess
16
  from PIL import Image
 
1
  from __future__ import annotations
2
 
3
+ import torch
4
+ torch.jit.script = lambda f: f
5
+
6
  import shlex
7
  import spaces
8
 
 
13
  import os
14
 
15
  from transformers import AutoModel, AutoProcessor
 
16
  import sys
17
  import subprocess
18
  from PIL import Image