ready to deploy

antoinekrajnc · antoinekrajnc · commit 0d4bd70e9326 · 2019-12-18T16:01:07.000+01:00
diff --git a/app.py b/app.py
@@ -0,0 +1,30 @@
+from flask import Flask, render_template, redirect, url_for, request, make_response, jsonify
+from sklearn.externals import joblib
+import numpy as np
+import requests
+import json
+
+app = Flask(__name__)
+
+
+@app.route("/")
+def index():
+    return render_template("index.html")
+
+@app.route("/predict", methods=['POST'])
+def predict():
+    if request.method=='POST':
+
+        regressor = joblib.load("linear_regression_model.pkl")
+
+        data = dict(request.form.items())
+
+        years_of_experience = np.array(float(data["YearsExperience"])).reshape(-1,1)
+
+        prediction = regressor.predict(years_of_experience)
+
+    return render_template("predicted.html", prediction=prediction)
+
+
+if __name__ == '__main__':
+    app.run(debug=True)
diff --git a/linear_regression_model.pkl b/linear_regression_model.pkl
diff --git a/prediction.py b/prediction.py
@@ -0,0 +1,6 @@
+import requests
+BASE_URL = "http://127.0.0.1:5000"
+years_exp = {"YearsExperience": 8}
+response = requests.post("{}/predict".format(BASE_URL), json = years_exp)
+
+print(response.json())
diff --git a/requirements.txt b/requirements.txt
@@ -0,0 +1,16 @@
+certifi==2019.11.28
+chardet==3.0.4
+Click==7.0
+Flask==1.1.1
+idna==2.8
+itsdangerous==1.1.0
+Jinja2==2.10.3
+joblib==0.14.1
+MarkupSafe==1.1.1
+numpy==1.17.4
+requests==2.22.0
+scikit-learn==0.22
+scipy==1.4.0
+sklearn==0.0
+urllib3==1.25.7
+Werkzeug==0.16.0
diff --git a/static/animations.js b/static/animations.js
diff --git a/static/style.css b/static/style.css
diff --git a/templates/index.html b/templates/index.html
@@ -0,0 +1,13 @@
+{% extends "layout.html" %}
+{% block content %}
+<!--Enter Name -->
+<div style="padding:10px;">
+  <div class="enter-name">
+    <form action="{{ url_for('predict') }}" method="POST">
+      <label>Année d'expérience</label>
+      <input type="integer" name="YearsExperience" autofocus>
+      <input type="submit" value="Let's predict salary!">
+    </form>
+  </div>
+</div>
+{% endblock %}
diff --git a/templates/layout.html b/templates/layout.html
@@ -0,0 +1,10 @@
+<!DOCTYPE html>
+<html>
+    <head>
+        <title>ML Project</title>
+        <meta name="viewport" content="width=device-width, initial-scale=1">
+    </head>
+    <body>
+        {% block content %}{% endblock %}
+    </body>
+</html>
diff --git a/templates/predicted.html b/templates/predicted.html
@@ -0,0 +1,9 @@
+{% extends "layout.html" %}
+{% block content %}
+<!--Enter Name -->
+<div class="enter-name">
+  <span>Voici le salaire prédit
+    <h1> {{ prediction| float }}</h1>
+  </span>
+</div>
+{% endblock %}
