Upload folder using huggingface_hub

main.py

@@ -1,4 +1,37 @@
 import gradio as gr
+import os
+import requests
+from tqdm import tqdm
+
+def check_and_download_model():
+    temp_dir = "/tmp"
+    model_path = os.path.join(temp_dir, "large-v2.pt")
+    
+    if os.path.exists(model_path):
+        return f"モデルは既に存在します: {model_path}"
+    
+    url = "https://openaipublic.azureedge.net/main/whisper/models/81f7c96c852ee8fc832187b0132e569d6c3065a3252ed18e56effd0b6a73e524/large-v2.pt"
+    
+    try:
+        response = requests.get(url, stream=True)
+        response.raise_for_status()
+        total_size = int(response.headers.get('content-length', 0))
+        
+        with open(model_path, 'wb') as f, tqdm(
+            desc=model_path,
+            total=total_size,
+            unit='iB',
+            unit_scale=True,
+            unit_divisor=1024,
+        ) as pbar:
+            for data in response.iter_content(chunk_size=1024):
+                size = f.write(data)
+                pbar.update(size)
+                
+        return f"モデルのダウンロードが完了しました: {model_path}"
+        
+    except Exception as e:
+        return f"エラーが発生しました: {e}"
 
 def click_test():
     """1から10までのランダムな数値を生成する関数"""
@@ -8,20 +41,34 @@ def click_test():
 
 # Gradio インターフェースの作成
 with gr.Blocks() as demo:
-    gr.Markdown("# ランダム数値ジェネレーター")
-    gr.Markdown("下のボタンをクリックすると1から10までのランダムな数値が生成されます。")
+    gr.Markdown("# Whisper モデルチェッカー & ランダム数値ジェネレーター")
+    gr.Markdown("下のボタンでモデルの確認・ダウンロードと、ランダム数値の生成ができます。")
     
     # 結果表示用のテキストボックス
-    output_text = gr.Text(label="結果")
+    model_status = gr.Text(label="モデルの状態")
+    output_text = gr.Text(label="ランダム数値の結果")
     
-    # カスタムボタンの追加
-    generate_btn = gr.Button(
-        value="数値を生成する",  # ボタンのテキスト
-        variant="primary",    # ボタンのスタイル
-        size="lg"            # ボタンのサイズ
-    )
+    with gr.Row():
+        # モデルチェック用ボタン
+        check_btn = gr.Button(
+            value="モデルを確認する",
+            variant="secondary",
+            size="lg"
+        )
+        
+        # ランダム数値生成用ボタン
+        generate_btn = gr.Button(
+            value="数値を生成する",
+            variant="primary",
+            size="lg"
+        )
     
     # ボタンクリック時のイベント設定
+    check_btn.click(
+        fn=check_and_download_model,
+        outputs=model_status
+    )
+    
     generate_btn.click(
         fn=click_test,
         outputs=output_text

speaker_pretrain/best_model.pth.tar

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b67178273ee7ea1cdea4378a3ec9a6e0da93d75238ad641ddb4714e4ef46ea14
+size 114064632