shmuelamar commited on
Commit
cf4739b
·
unverified ·
1 Parent(s): 5c2e272

try to fix spaces RuntimeError due to cuda initialized before spaces

Browse files
Files changed (1) hide show
  1. app.py +6 -5
app.py CHANGED
@@ -5,6 +5,12 @@ import logging
5
  import os
6
  from pathlib import Path
7
 
 
 
 
 
 
 
8
  import gradio as gr
9
  import tiktoken
10
  import torch
@@ -16,11 +22,6 @@ from igcs.entities import Doc, Selection
16
  from igcs.utils import log
17
  from igcs.utils.diskcache import disk_cache
18
 
19
- try:
20
- import spaces
21
- except ModuleNotFoundError:
22
- spaces = lambda: None
23
- spaces.GPU = lambda fn: fn
24
 
25
  logger = logging.getLogger("igcs-demo")
26
 
 
5
  import os
6
  from pathlib import Path
7
 
8
+ try:
9
+ import spaces
10
+ except ModuleNotFoundError:
11
+ spaces = lambda: None
12
+ spaces.GPU = lambda fn: fn
13
+
14
  import gradio as gr
15
  import tiktoken
16
  import torch
 
22
  from igcs.utils import log
23
  from igcs.utils.diskcache import disk_cache
24
 
 
 
 
 
 
25
 
26
  logger = logging.getLogger("igcs-demo")
27