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)

Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support