Spaces:
Sleeping
Sleeping
Update app.py
Browse filesSwitch to facebook/bart-large-cnn due to 503 error
app.py
CHANGED
@@ -10,19 +10,16 @@ if not HF_TOKEN:
|
|
10 |
print(f"HF_TOKEN loaded: {len(HF_TOKEN) if HF_TOKEN else 0} characters")
|
11 |
|
12 |
try:
|
13 |
-
summary_client = InferenceClient(model="
|
14 |
translation_client = InferenceClient(model="facebook/m2m100_418M", token=HF_TOKEN)
|
15 |
def generate_summary(text):
|
16 |
for _ in range(3):
|
17 |
try:
|
18 |
-
# 首先生成英文摘要
|
19 |
response = summary_client.summarization(text)
|
20 |
print(f"Summary API Response: {response}")
|
21 |
summary_text = response.generated_text if hasattr(response, 'generated_text') and response.generated_text else response.summary_text if hasattr(response, 'summary_text') and response.summary_text else None
|
22 |
if not summary_text:
|
23 |
return "摘要生成失败,属性缺失。"
|
24 |
-
|
25 |
-
# 将英文摘要翻译为中文
|
26 |
translated_response = translation_client.text_generation(
|
27 |
summary_text,
|
28 |
src_lang="en",
|
@@ -35,7 +32,8 @@ try:
|
|
35 |
error_details = traceback.format_exc()
|
36 |
print(f"尝试失败,错误类型: {type(e).__name__}, 错误详情: {str(e)}, 堆栈: {error_details}")
|
37 |
time.sleep(1)
|
38 |
-
|
|
|
39 |
interface = gr.Interface(
|
40 |
fn=generate_summary,
|
41 |
inputs=gr.Textbox(lines=5, placeholder="输入文档内容..."),
|
|
|
10 |
print(f"HF_TOKEN loaded: {len(HF_TOKEN) if HF_TOKEN else 0} characters")
|
11 |
|
12 |
try:
|
13 |
+
summary_client = InferenceClient(model="facebook/bart-large-cnn", token=HF_TOKEN)
|
14 |
translation_client = InferenceClient(model="facebook/m2m100_418M", token=HF_TOKEN)
|
15 |
def generate_summary(text):
|
16 |
for _ in range(3):
|
17 |
try:
|
|
|
18 |
response = summary_client.summarization(text)
|
19 |
print(f"Summary API Response: {response}")
|
20 |
summary_text = response.generated_text if hasattr(response, 'generated_text') and response.generated_text else response.summary_text if hasattr(response, 'summary_text') and response.summary_text else None
|
21 |
if not summary_text:
|
22 |
return "摘要生成失败,属性缺失。"
|
|
|
|
|
23 |
translated_response = translation_client.text_generation(
|
24 |
summary_text,
|
25 |
src_lang="en",
|
|
|
32 |
error_details = traceback.format_exc()
|
33 |
print(f"尝试失败,错误类型: {type(e).__name__}, 错误详情: {str(e)}, 堆栈: {error_details}")
|
34 |
time.sleep(1)
|
35 |
+
return f"网络错误,请稍后重试。错误详情:{str(e)}"
|
36 |
+
return "网络错误,请稍后重试(重试次数已达上限)。"
|
37 |
interface = gr.Interface(
|
38 |
fn=generate_summary,
|
39 |
inputs=gr.Textbox(lines=5, placeholder="输入文档内容..."),
|