doropiza commited on
Commit
bf4d917
Β·
1 Parent(s): 8f61896
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -86,7 +86,7 @@ class TextSummarizer:
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
  )
 
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
  )