Update app.py

app.py

@@ -30,7 +30,8 @@ def convert_and_deploy(url, repo_id, hf_token):
     # モデルを読み込み
     try:
         # モデルの構成を取得
-        config = AutoConfig.from_pretrained("Sakalti/Qwen-float16-0.5b", token=hf_token)
+        model_name = repo_id.split('/')[-1]  # モデル名を取得
+        config = AutoConfig.from_pretrained(model_name, token=hf_token)
         
         # モデルを構成に基づいて初期化
         model = AutoModel.from_config(config, torch_dtype=torch.float16)
@@ -49,7 +50,6 @@ def convert_and_deploy(url, repo_id, hf_token):
     
     # モデルをfloat16形式で保存
     try:
-        model_name = repo_id.split('/')[-1]  # モデル名を取得
         model.save_pretrained(f"{model_name}_float16", torch_dtype=torch.float16)
     except Exception as e:
         return f"モデルの保存に失敗しました。エラー: {e}"