import requests from flask import Flask, request, jsonify
app = Flask(name)
Hugging Face API Key (Replace with your actual API key)
HF_API_KEY = "your_huggingface_api_key" MODEL_URL = "https://api-inference.huggingface.co/models/gpt2"
Function to query Hugging Face API
def query_huggingface(text): headers = {"Authorization": f"Bearer {HF_API_KEY}"} data = {"inputs": text}
try:
response = requests.post(MODEL_URL, headers=headers, json=data, timeout=10)
response.raise_for_status()
return response.json()
except requests.exceptions.RequestException as e:
return {"error": str(e)}
API Route
@app.route('/generate', methods=['POST']) def generate_text(): user_input = request.json.get("text", "") if not user_input: return jsonify({"error": "No text provided"}), 400
response = query_huggingface(user_input)
return jsonify(response)
if name == 'main': app.run(host='0.0.0.0', port=7860)