Spaces:
Runtime error
Runtime error
freemt
commited on
Commit
·
362de3f
1
Parent(s):
1f95340
Update
Browse files- .flake8 +14 -0
- .gitignore +1 -0
- app.py +30 -11
- gradio_queue.db +0 -0
- pyproject.toml +8 -0
.flake8
ADDED
@@ -0,0 +1,14 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
[flake8]
|
2 |
+
# E503: line break before binary operator, black's format style
|
3 |
+
ignore = D203,E501,W503
|
4 |
+
exclude =
|
5 |
+
.git,
|
6 |
+
__pycache__,
|
7 |
+
docs/source/conf.py,
|
8 |
+
old,
|
9 |
+
build,
|
10 |
+
dist,
|
11 |
+
.venv,
|
12 |
+
pad*.py
|
13 |
+
max-line-length = 120
|
14 |
+
max-complexity = 30
|
.gitignore
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
gradio_queue.db
|
app.py
CHANGED
@@ -1,19 +1,19 @@
|
|
1 |
"""See https://huggingface.co/spaces/Gradio-Blocks/Story-to-video/blob/main/app.py."""
|
2 |
import base64
|
3 |
import io
|
|
|
4 |
import os
|
5 |
import re
|
6 |
import time
|
7 |
-
from random import choice
|
8 |
|
9 |
import gradio as gr
|
10 |
import translators as ts
|
11 |
from fastlid import fastlid
|
12 |
from logzero import logger
|
13 |
from PIL import Image # opencv-python
|
14 |
-
from random import choices
|
15 |
from tenacity import retry
|
16 |
-
from tenacity.stop import stop_after_attempt
|
17 |
|
18 |
# from PIL import Image
|
19 |
# from transformers import AutoTokenizer, AutoModelForSeq2SeqLM,pipeline
|
@@ -29,11 +29,14 @@ try:
|
|
29 |
except Exception:
|
30 |
... # Windows
|
31 |
|
|
|
|
|
|
|
32 |
|
33 |
examples_ = [
|
34 |
-
"黑云压城城欲摧 ,甲光向日金鳞开。",
|
35 |
"黄金在河里流淌,宝石遍地,空中铺满巨大的彩虹。",
|
36 |
"蓝色的夜,森林中好多萤火虫",
|
|
|
37 |
"季姬寂,集鸡,鸡即棘鸡。棘鸡饥叽,季姬及箕稷济鸡。",
|
38 |
"an apple",
|
39 |
"a cat",
|
@@ -42,9 +45,13 @@ examples_ = [
|
|
42 |
]
|
43 |
|
44 |
|
45 |
-
# @retry(stop=
|
46 |
-
@retry(stop=stop_after_attempt(5))
|
47 |
-
def tr_(
|
|
|
|
|
|
|
|
|
48 |
"""Wrap [ts.deepl, ts.baidu, ts.google] with tenacity.
|
49 |
|
50 |
not working: sogou; ?tencent
|
@@ -52,14 +59,18 @@ def tr_(text: str) -> str:
|
|
52 |
cand = [ts.baidu, ts.youdao, ts.google]
|
53 |
for tr in [ts.deepl] + choices(cand, k=len(cand)):
|
54 |
try:
|
55 |
-
res = tr(
|
|
|
|
|
|
|
|
|
56 |
logger.info(" api used: %s", tr.__name__)
|
57 |
tr_.api_used = tr.__name__
|
58 |
break
|
59 |
except Exception:
|
60 |
continue
|
61 |
else:
|
62 |
-
res = "Something is probably
|
63 |
|
64 |
return res
|
65 |
|
@@ -76,13 +87,21 @@ def generate_images(phrase: str, steps: int = 125):
|
|
76 |
except Exception as exc:
|
77 |
logger.error(exc)
|
78 |
|
|
|
|
|
|
|
79 |
# safeguard short Chinese phrases
|
80 |
-
if len(phrase)
|
81 |
detected = "zh"
|
|
|
82 |
|
83 |
if detected not in ["en"]:
|
84 |
try:
|
85 |
-
generated_text = tr_(
|
|
|
|
|
|
|
|
|
86 |
extra_info = f"({tr_.api_used}: {generated_text})"
|
87 |
except Exception as exc:
|
88 |
logger.error(exc)
|
|
|
1 |
"""See https://huggingface.co/spaces/Gradio-Blocks/Story-to-video/blob/main/app.py."""
|
2 |
import base64
|
3 |
import io
|
4 |
+
import logzero
|
5 |
import os
|
6 |
import re
|
7 |
import time
|
8 |
+
from random import choice, choices
|
9 |
|
10 |
import gradio as gr
|
11 |
import translators as ts
|
12 |
from fastlid import fastlid
|
13 |
from logzero import logger
|
14 |
from PIL import Image # opencv-python
|
|
|
15 |
from tenacity import retry
|
16 |
+
from tenacity.stop import stop_after_attempt, stop_after_delay
|
17 |
|
18 |
# from PIL import Image
|
19 |
# from transformers import AutoTokenizer, AutoModelForSeq2SeqLM,pipeline
|
|
|
29 |
except Exception:
|
30 |
... # Windows
|
31 |
|
32 |
+
loglevel = 10 # change to 20 to switch off debug
|
33 |
+
logzero.loglevel(loglevel)
|
34 |
+
|
35 |
|
36 |
examples_ = [
|
|
|
37 |
"黄金在河里流淌,宝石遍地,空中铺满巨大的彩虹。",
|
38 |
"蓝色的夜,森林中好多萤火虫",
|
39 |
+
"黑云压城城欲摧 ,甲光向日金鳞开。",
|
40 |
"季姬寂,集鸡,鸡即棘鸡。棘鸡饥叽,季姬及箕稷济鸡。",
|
41 |
"an apple",
|
42 |
"a cat",
|
|
|
45 |
]
|
46 |
|
47 |
|
48 |
+
# @retry(stop=stop_after_attempt(5))
|
49 |
+
@retry(stop=(stop_after_delay(10) | stop_after_attempt(5)))
|
50 |
+
def tr_(
|
51 |
+
text: str,
|
52 |
+
from_language="auto",
|
53 |
+
to_language="en",
|
54 |
+
) -> str:
|
55 |
"""Wrap [ts.deepl, ts.baidu, ts.google] with tenacity.
|
56 |
|
57 |
not working: sogou; ?tencent
|
|
|
59 |
cand = [ts.baidu, ts.youdao, ts.google]
|
60 |
for tr in [ts.deepl] + choices(cand, k=len(cand)):
|
61 |
try:
|
62 |
+
res = tr(
|
63 |
+
text,
|
64 |
+
from_language=from_language,
|
65 |
+
to_language=to_language,
|
66 |
+
)
|
67 |
logger.info(" api used: %s", tr.__name__)
|
68 |
tr_.api_used = tr.__name__
|
69 |
break
|
70 |
except Exception:
|
71 |
continue
|
72 |
else:
|
73 |
+
res = "Something is probably wrong, ping dev to fix it if you like."
|
74 |
|
75 |
return res
|
76 |
|
|
|
87 |
except Exception as exc:
|
88 |
logger.error(exc)
|
89 |
|
90 |
+
logzero.loglevel(loglevel)
|
91 |
+
logger.debug("phrase: %s, deteted: %s", phrase, detected)
|
92 |
+
|
93 |
# safeguard short Chinese phrases
|
94 |
+
if len(phrase) <= 10 and re.search(r"[一-龟]+", phrase):
|
95 |
detected = "zh"
|
96 |
+
logger.debug(" safeguard branch ")
|
97 |
|
98 |
if detected not in ["en"]:
|
99 |
try:
|
100 |
+
generated_text = tr_(
|
101 |
+
phrase,
|
102 |
+
detected,
|
103 |
+
"en",
|
104 |
+
)
|
105 |
extra_info = f"({tr_.api_used}: {generated_text})"
|
106 |
except Exception as exc:
|
107 |
logger.error(exc)
|
gradio_queue.db
CHANGED
Binary files a/gradio_queue.db and b/gradio_queue.db differ
|
|
pyproject.toml
CHANGED
@@ -34,3 +34,11 @@ export = "poetry export --without-hashes -f requirements.txt -o requirements.txt
|
|
34 |
[tool.isort]
|
35 |
profile = "black"
|
36 |
multi_line_output = 3
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
34 |
[tool.isort]
|
35 |
profile = "black"
|
36 |
multi_line_output = 3
|
37 |
+
|
38 |
+
[tool.pyright]
|
39 |
+
venvPath = ".venv"
|
40 |
+
reportTypeshedErrors = false
|
41 |
+
reportMissingImports = true
|
42 |
+
reportMissingTypeStubs = false
|
43 |
+
pythonVersion = "3.8"
|
44 |
+
ignore = []
|