| ''' |
| Author : Alafun |
| GitHub Page : https://github.com/Alafun/ |
| Date : 2022-06-21 16:43:21 |
| LastEditors : Alafun |
| LastEditTime : 2022-06-21 18:51:37 |
| Description : |
| FilePath : \\app.py |
| Copyright (c) 2022 by Alafun, All Rights Reserved. |
| ''' |
|
|
|
|
| import gradio as gr |
|
|
|
|
| def sepia(input_img): |
| sepia_filter = np.array( |
| [[0.393, 0.769, 0.189], [0.349, 0.686, 0.168], [0.272, 0.534, 0.131]] |
| ) |
| sepia_img = input_img.dot(sepia_filter.T) |
| sepia_img /= sepia_img.max() |
| return sepia_img |
|
|
| import tensorflow as tf |
| import numpy as np |
| from urllib.request import urlretrieve |
|
|
|
|
|
|
| urlretrieve("https://gr-models.s3-us-west-2.amazonaws.com/mnist-model.h5", "mnist-model.h5") |
| model = tf.keras.models.load_model("mnist-model.h5") |
|
|
| def recognize_digit(image): |
| image = image.reshape(1, -1) |
| prediction = model.predict(image).tolist()[0] |
| return {str(i): prediction[i] for i in range(10)} |
| |
| TITLE="RECOULLY: RECOgnize yoUr caLLigraphY" |
| description="""在手写区写出你心里想的以为数字(0~9), 查看小美同学的实时预测吧! |
| |
| ### 项目介绍 |
| - 中文名: 小美同学 |
| - 英文名: Recoully |
| - 口头禅: 小美这么可爱, 就算猜错了也希望被原谅~~qaq |
| |
| """ |
| demo = gr.Interface(fn=recognize_digit, |
| inputs="sketchpad", |
| outputs=gr.outputs.Label(num_top_classes=3), |
| live=True, |
| css=".footer {display:none !important}", |
| title=TITLE, |
| description=description, |
| thumbnail="https://raw.githubusercontent.com/gradio-app/real-time-mnist/master/thumbnail2.png") |
|
|
| demo.launch() |
|
|
