Currency-Authenticator-Using-ML/flask_main.py at main · gauravabhishek4U/Currency-Authenticator-Using-ML · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
# -*- coding: utf-8 -*-
"""
Created on Sun Jan 3, 14:59:31 2021

@author: Abhishek Gaurav
"""

from flask import Flask, request
import pickle
import pandas as pd
from flasgger import Swagger

app = Flask(__name__)
Swagger(app)

pickle_in = open('classifier.pkl','rb')
classifier = pickle.load(pickle_in)

@app.route('/')
def Welcome():
    return 'Welcome Everyone'

@app.route('/predict', methods=["GET"])
def predict_note_authentication():

    """Let's Authenticate the Paper Currency
    Built by ABHISHEK GAURAV.
    ---
    parameters:
      - name: variance
        in: query
        type: number
        required: true
      - name: skewness
        in: query
        type: number
        required: true
      - name: curtosis
        in: query
        type: number
        required: true
      - name: entropy
        in: query
        type: number
        required: true
    responses:
        200:
            description: The output value is displayed in Response Body.

    """

    variance = request.args.get('variance')
    skewness = request.args.get('skewness')
    curtosis = request.args.get('curtosis')
    entropy = request.args.get('entropy')
    prediction = classifier.predict([[variance,skewness,curtosis,entropy]])
    return "The Predicted Value is "+str(prediction)

@app.route('/predict_file',methods=['POST'])
def predict_note_file():

    """Let's Authenticate the Paper Currency
    Built by ABHISHEK GAURAV.
    ---
    parameters:
      - name: file
        in: formData
        type: file
        required: true

    responses:
        200:
            description: The output values is displayed in Response Body.

    """

    df_test = pd.read_csv(request.files.get("file"))
    prediction = classifier.predict(df_test)
    return "The Predicted values for the test data is "+str(list(prediction))


if __name__=='__main__':
    app.run()