BeveledCube/bevelapi

1from flask import Flask, request, render_template, jsonify2from models import gpt2 as chatbot3 4app = Flask("AI API")5 6print("Loading model")7chatbot.load()8 9@app.get("/")10def read_root():11  return render_template("index.html")12 13@app.route("/test")14def test_route():15  return "This is a test route."16 17@app.route("/api", methods=["POST"])18def receive_data():19  data = request.get_json()20  print("Prompt:", data["prompt"])21  22  generated_text = chatbot.generate(data["prompt"])23  24  print("Response:", generated_text)25  26  return jsonify({ "answer": generated_text })27 28app.run(host="0.0.0.0", port=7860, debug=False)