Yehor commited on
Commit
cecf79d
·
1 Parent(s): 8a6f9a8

Fix spaces init

Browse files
Files changed (1) hide show
  1. app.py +10 -3
app.py CHANGED
@@ -4,8 +4,14 @@ import json
4
  import time
5
 
6
  from importlib.metadata import version
 
7
  from enum import Enum
8
 
 
 
 
 
 
9
  import torch
10
  import torchaudio
11
 
@@ -26,11 +32,11 @@ from vocos import Vocos
26
  use_zerogpu = False
27
 
28
  try:
29
- import spaces # it's for ZeroGPU
30
-
31
  use_zerogpu = True
32
  print("ZeroGPU is available, changing inference call.")
33
- except ImportError:
 
34
  print("ZeroGPU is not available, skipping...")
35
 
36
 
@@ -142,6 +148,7 @@ tech_libraries = f"""
142
  #### Libraries
143
 
144
  - gradio: {version("gradio")}
 
145
  - torch: {version("torch")}
146
  - scipy: {version("scipy")}
147
  - numba: {version("numba")}
 
4
  import time
5
 
6
  from importlib.metadata import version
7
+ from importlib.metadata import PackageNotFoundError
8
  from enum import Enum
9
 
10
+ try:
11
+ import spaces # it's for ZeroGPU
12
+ except ImportError:
13
+ ...
14
+
15
  import torch
16
  import torchaudio
17
 
 
32
  use_zerogpu = False
33
 
34
  try:
35
+ spaces_version = version("spaces")
 
36
  use_zerogpu = True
37
  print("ZeroGPU is available, changing inference call.")
38
+ except PackageNotFoundError:
39
+ spaces_version = "N/A"
40
  print("ZeroGPU is not available, skipping...")
41
 
42
 
 
148
  #### Libraries
149
 
150
  - gradio: {version("gradio")}
151
+ - spaces: {spaces_version}
152
  - torch: {version("torch")}
153
  - scipy: {version("scipy")}
154
  - numba: {version("numba")}