doropiza commited on
Commit
b952cb6
·
1 Parent(s): bf4d917
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -62,7 +62,7 @@ class TextSummarizer:
62
 
63
  # ZeroGPU環境では軽量モデルを優先
64
  # if self.is_zerogpu:
65
- model_name = "rinna/japanese-gpt2-medium"
66
  # else:
67
  # model_name = "facebook/bart-large-cnn"
68
 
@@ -86,7 +86,7 @@ class TextSummarizer:
86
  print("最軽量フォールバックモデルを試行中...")
87
  self.summarizer = pipeline(
88
  "summarization",
89
- model=model_name,
90
  device=0 if self.device == "cuda" else -1,
91
  trust_remote_code=False
92
  )
 
62
 
63
  # ZeroGPU環境では軽量モデルを優先
64
  # if self.is_zerogpu:
65
+ model_name = "sshleifer/distilbart-cnn-12-6"
66
  # else:
67
  # model_name = "facebook/bart-large-cnn"
68
 
 
86
  print("最軽量フォールバックモデルを試行中...")
87
  self.summarizer = pipeline(
88
  "summarization",
89
+ model="sshleifer/distilbart-cnn-6-6",
90
  device=0 if self.device == "cuda" else -1,
91
  trust_remote_code=False
92
  )