doropiza commited on
Commit
16ee848
·
1 Parent(s): 10df927
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -61,10 +61,10 @@ class TextSummarizer:
61
  print("モデルを読み込み中...")
62
 
63
  # ZeroGPU環境では軽量モデルを優先
64
- if self.is_zerogpu:
65
- model_name = "cyberagent/open-calm-7b" # より軽量
66
- else:
67
- model_name = "facebook/bart-large-cnn"
68
 
69
  # モデル読み込み設定(use_safetensorsを削除)
70
  pipeline_kwargs = {
 
61
  print("モデルを読み込み中...")
62
 
63
  # ZeroGPU環境では軽量モデルを優先
64
+ # if self.is_zerogpu:
65
+ model_name = "cyberagent/open-calm-7b"
66
+ # else:
67
+ # model_name = "facebook/bart-large-cnn"
68
 
69
  # モデル読み込み設定(use_safetensorsを削除)
70
  pipeline_kwargs = {