import os | |
import requests | |
HF_TOKEN = os.getenv("HF_TOKEN") | |
API_URL = "https://api-inference.huggingface.co/models/microsoft/DialoGPT-medium" | |
headers = {"Authorization": f"Bearer {HF_TOKEN}"} | |
def get_bot_response(user_input): | |
payload = {"inputs": {"text": user_input}} | |
response = requests.post(API_URL, headers=headers, json=payload) | |
if response.status_code != 200: | |
return f"β Error: {response.status_code} - {response.json()}" | |
return response.json()[0]["generated_text"] | |