commit

app.py

@@ -2,16 +2,12 @@ import gradio as gr
 import torch
 import os
 from transformers import AutoTokenizer, AutoModelForCausalLM
-from huggingface_hub import spaces
 
 # 環境変数からトークンを取得
 HUGGINGFACE_TOKEN = os.getenv("HUGGINGFACE_TOKEN")
 if not HUGGINGFACE_TOKEN:
     raise ValueError("HUGGINGFACE_TOKEN environment variable is not set")
 
-@spaces.GPU
-def init():
-    pass
 
 def _load_model():
     if not torch.cuda.is_available():