jinv2 commited on
Commit
d9fa0df
·
verified ·
1 Parent(s): e4ec2ca

Update app.py

Browse files

Switch to facebook/bart-large-cnn due to 503 error

Files changed (1) hide show
  1. app.py +3 -5
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="google/pegasus-xsum", 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
- # 首先生成英文摘要
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
- return "网络错误,请稍后重试。"
 
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="输入文档内容..."),