commit

app.py

@@ -9,9 +9,13 @@ HUGGINGFACE_TOKEN = os.getenv("HUGGINGFACE_TOKEN")
 if not HUGGINGFACE_TOKEN:
     raise ValueError("HUGGINGFACE_TOKEN environment variable is not set")
 
-@spaces.GPU
-def setup():
-    pass
+def _load_model():
+    if not torch.cuda.is_available():
+        raise RuntimeError("GPU is not available but required.")
+    print("GPU is available and model will be loaded.")
+    return "GPU ready"
+
+_load_model()
 
 # モデルとトークナイザーの初期化
 MODEL_NAME = "google/gemma-7b-it"