Spaces:
Sleeping
Sleeping
commit
Browse files
app.py
CHANGED
@@ -62,7 +62,7 @@ class TextSummarizer:
|
|
62 |
|
63 |
# ZeroGPU環境では軽量モデルを優先
|
64 |
# if self.is_zerogpu:
|
65 |
-
model_name = "
|
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=
|
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 |
)
|