Snigs98's picture
Upload app.py
649f75e verified
raw
history blame contribute delete
813 Bytes
from flask import Flask, request, jsonify, send_file
import pickle
import random
app = Flask(__name__)
model = pickle.load(open("model.pkl","rb"))
vectorizer = pickle.load(open("vectorizer.pkl","rb"))
intents = pickle.load(open("intents.pkl","rb"))
def get_response(message):
X = vectorizer.transform([message])
tag = model.predict(X)[0]
for intent in intents["intents"]:
if intent["tag"] == tag:
return random.choice(intent["responses"])
@app.route("/")
def home():
return send_file("index.html")
@app.route("/chat", methods=["POST"])
def chat():
user_message = request.json["message"]
response = get_response(user_message)
return jsonify({"response": response})
if __name__ == "__main__":
app.run(host="0.0.0.0", port=7860)