Update app.py

app.py

@@ -1,12 +1,14 @@
 import gradio as gr
 from transformers import pipeline
-import spaces
+import os
+
+# Hugging Faceのアクセストークンを環境変数から取得
+access_token = os.getenv("HUGGING_FACE_HUB_TOKEN")
 
 # SakanaAIのテキスト生成モデルの初期化
-text_generator = pipeline("text-generation", model="SakanaAI/your-model-name")
+text_generator = pipeline("text-generation", model="SakanaAI/your-model-name", use_auth_token=access_token)
 
 # GPUを使用する関数の定義
-@spaces.GPU
 def generate_text(prompt):
     result = text_generator(prompt, max_length=50)[0]
     return result['generated_text']