Commit
·
ae91ef9
1
Parent(s):
a70eba7
fix
Browse files
webui.py
CHANGED
|
@@ -1,3 +1,4 @@
|
|
|
|
|
| 1 |
import json
|
| 2 |
import os
|
| 3 |
import sys
|
|
@@ -100,6 +101,7 @@ def normalize_emo_vec(emo_vec):
|
|
| 100 |
tmp = tmp * 0.8/ np.sum(tmp)
|
| 101 |
return tmp.tolist()
|
| 102 |
|
|
|
|
| 103 |
def gen_single(emo_control_method,prompt, text,
|
| 104 |
emo_ref_path, emo_weight,
|
| 105 |
vec1, vec2, vec3, vec4, vec5, vec6, vec7, vec8,
|
|
|
|
| 1 |
+
import spaces
|
| 2 |
import json
|
| 3 |
import os
|
| 4 |
import sys
|
|
|
|
| 101 |
tmp = tmp * 0.8/ np.sum(tmp)
|
| 102 |
return tmp.tolist()
|
| 103 |
|
| 104 |
+
@spaces.GPU
|
| 105 |
def gen_single(emo_control_method,prompt, text,
|
| 106 |
emo_ref_path, emo_weight,
|
| 107 |
vec1, vec2, vec3, vec4, vec5, vec6, vec7, vec8,
|