Tag: Hugging Face

  • Hugging Face APIの導通確認

    Hugging face APIを使用して,LLMの応答試験を行います. ドラフトのコードをChatGPTで生成し,動作するように修正しました. from huggingface_hub import InferenceClient import os from dotenv import load_dotenv # .envファイルから環境変数を読み込む(HF_TOKENを保存) load_dotenv() client = InferenceClient(token=os.getenv(“HF_TOKEN”)) # 1. 英語のテキストでテスト print(“=== 英語テキストのテスト ===”) try: response = client.text_generation(“Hello, world!”, model=”gpt2″) print(“レスポンスの内容:”, repr(response)) except Exception as e: print(“エラーが発生しました:”, str(e)) # 2. 日本語対応モデルでテスト print(“\n=== 日本語対応モデルのテスト ===”) try: response = client.text_generation( “こんにちは、世界!”, model=”rinna/japanese-gpt2-medium” # 日本語対応GPT-2モデル )…