|
|
import os |
|
|
os.system("pip uninstall -y gradio") |
|
|
os.system("pip install gradio==3.50.2") |
|
|
import tensorflow as tf |
|
|
from matplotlib import pyplot as plt |
|
|
import numpy as np |
|
|
import gradio as gr |
|
|
|
|
|
|
|
|
model = tf.keras.models.load_model('handwritten_digits.model') |
|
|
|
|
|
def recognize_digit(image): |
|
|
if image is not None: |
|
|
image = image.reshape((1,28,28,1)).astype('float32')/255 |
|
|
prediction = model.predict(image) |
|
|
return {str(i) : float(prediction[0][i]) for i in range(10)} |
|
|
else: |
|
|
return '' |
|
|
|
|
|
iface = gr.Interface( |
|
|
fn = recognize_digit, |
|
|
inputs=gr.Image(shape=(28,28),image_mode = 'L',invert_colors=True, source = 'canvas'), |
|
|
outputs=gr.Label(top_num_classes=3)) |
|
|
|
|
|
iface.launch() |
