Spaces:
Runtime error
Runtime error
File size: 2,682 Bytes
7199111 8ecf185 7199111 8ecf185 3fac891 d977393 8ecf185 4108613 f58a881 d977393 ba89bc9 4108613 088b445 4108613 088b445 4108613 f58a881 4108613 ba89bc9 f58a881 3657998 f58a881 4108613 9eb9696 4108613 ba89bc9 088b445 4108613 088b445 4108613 8ecf185 9d9428d bbcce29 4108613 9eb9696 bbcce29 |
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 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 |
from flask import *
from PIL import Image
import face_recognition
import cv2
import numpy as np
import csv
from datetime import datetime
############################################
#################
from flask_socketio import SocketIO,emit
import base64
##################
app = Flask (__name__ )
#################
app.config['SECRET_KEY'] = 'secret!'
socket = SocketIO(app,async_mode="eventlet")
#######################
######################
def base64_to_image(base64_string):
# Extract the base64 encoded binary data from the input string
base64_data = base64_string.split(",")[1]
# Decode the base64 data to bytes
image_bytes = base64.b64decode(base64_data)
# Convert the bytes to numpy array
image_array = np.frombuffer(image_bytes, dtype=np.uint8)
# Decode the numpy array as an image using OpenCV
image = cv2.imdecode(image_array, cv2.IMREAD_COLOR)
return image
@socket.on("connect")
def test_connect():
print("Connected")
emit("my response", {"data": "Connected"})
@socket.on("image")
def receive_image(image):
# Decode the base64-encoded image data
image = base64_to_image(image)
image = cv2.resize(image, (224, 224), interpolation=cv2.INTER_AREA)
# emit("processed_image", image)
# Make the image a numpy array and reshape it to the models input shape.
image = np.asarray(image, dtype=np.float32).reshape(1, 224, 224, 3)
image = (image / 127.5) - 1
# Predicts the model
prediction = model.predict(image)
index = np.argmax(prediction)
class_name = class_names[index]
confidence_score = prediction[0][index]
emit("result",{"name":str(class_name),"score":str(confidence_score)})
#######################
@app.route ("/")
def home():
return render_template("index.html")
if __name__ == '__main__':
socket.run(app,host="0.0.0.0", port=7860)
###########################################################################
# @app.route('/table')
# def show_table():
# # Get the current date
# current_date = datetime.now().strftime("%Y-%m-%d")
# # Read the CSV file to get attendance data
# attendance=[]
# try:
# with open(f"{current_date}.csv", newline="") as csv_file:
# csv_reader = csv.reader(csv_file)
# attendance = list(csv_reader)
# except FileNotFoundError:
# pass
# # Render the table.html template and pass the attendance data
# return render_template('attendance.html', attendance=attendance)
# @app.route("/")
# def home():
# return render_template('index.html')
# if __name__ == "__main__":
# socket.run(app,host="0.0.0.0", port=7860)
|