commit

app.py

@@ -8,20 +8,10 @@ Hugging Face Spaces (ZeroGPU) 対応版
 - openai/gpt-oss-20b
 """
 
-import gradio as gr
-import torch
-from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 import os
 from typing import List, Tuple
 
-# Hugging Face token from environment variable
-HF_TOKEN = os.getenv("HF_TOKEN")
-
-# トークンのチェック
-if not HF_TOKEN:
-    print("警告: HF_TOKENが設定されていません。プライベートモデルへのアクセスが制限される場合があります。")
-
-# Check if running on ZeroGPU
+# Check if running on ZeroGPU FIRST (before any CUDA initialization)
 try:
     import spaces
     IS_ZEROGPU = True
@@ -30,6 +20,18 @@ except ImportError:
     IS_ZEROGPU = False
     print("通常のGPU/CPU環境で実行しています。")
 
+# Import after spaces check
+import gradio as gr
+import torch
+from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
+
+# Hugging Face token from environment variable
+HF_TOKEN = os.getenv("HF_TOKEN")
+
+# トークンのチェック
+if not HF_TOKEN:
+    print("警告: HF_TOKENが設定されていません。プライベートモデルへのアクセスが制限される場合があります。")
+
 class ChatBot:
     def __init__(self):
         self.model = None