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="输入文档内容..."),
|