tamatwi commited on
Commit
af5341b
·
verified ·
1 Parent(s): b8e6e93

Update app.py

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