|
''' |
|
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)} |
|
|
|
demo = gr.Interface(fn=recognize_digit, |
|
inputs="sketchpad", |
|
outputs=gr.outputs.Label(num_top_classes=3), |
|
live=True, |
|
css=".footer {display:none !important}", |
|
|
|
description="Draw a number 0 through 9 on the sketchpad, and see predictions in real time.", |
|
thumbnail="https://raw.githubusercontent.com/gradio-app/real-time-mnist/master/thumbnail2.png") |
|
|
|
demo.launch() |
|
|
