Shuu12121 commited on
Commit
569f6b9
·
verified ·
1 Parent(s): d821efc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ from spaces import GPU
9
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
10
  print(f"Using device: {device}") # デバイス確認用ログ
11
 
12
- model_name = "Shuu12121/CodeEncoderDecoderModel-Ghost"
13
  print(f"Loading model: {model_name}") # モデル読み込み開始ログ
14
 
15
  # --- Tokenizerの読み込み ---
 
9
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
10
  print(f"Using device: {device}") # デバイス確認用ログ
11
 
12
+ model_name = "Shuu12121/CodeEncoderDecoderModel-Ghost-large"
13
  print(f"Loading model: {model_name}") # モデル読み込み開始ログ
14
 
15
  # --- Tokenizerの読み込み ---