import requests | |
import os | |
# Hugging Face APIエンドポイントと認証トークン | |
api_url = "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.2" | |
headers = {"Authorization": f"Bearer {os.getenv('HF_TOKEN')}"} | |
# クエリのメッセージ | |
payload = { | |
"inputs": "Which of the depicted countries has the best food?", | |
"parameters": {"max_new_tokens": 512} | |
} | |
# APIリクエスト | |
response = requests.post(api_url, headers=headers, json=payload) | |
# 結果表示 | |
if response.status_code == 200: | |
print("Response:", response.json()) | |
else: | |
print(f"Error {response.status_code}: {response.text}") | |