File size: 925 Bytes
aad3cee
0af5b3f
aad3cee
22c29e4
0af5b3f
aad3cee
 
 
 
fb78c6b
aad3cee
 
 
 
 
 
 
 
 
fb78c6b
aad3cee
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
# gr.Interface.load("models/PranomVignesh/Handwritten-Characters").launch()
import gradio as gr
from transformers import pipeline
from PIL import Image

imageClassifier = pipeline(task="image-classification",
                           model="PranomVignesh/Handwritten-Characters")

def predict(image):
    image = Image.fromarray(image)
    predictions = imageClassifier(image)
    output = {}
    for item in predictions:
        output[item['label']] = item['score']

    return output


inputs = gr.Sketchpad(shape=(224, 224),
                  live=True,
                  label="Draw your characters to detect")

outputs = gr.Label(label="Predicted Character")

interface = gr.Interface(
    fn=predict,
    inputs=inputs,
    outputs=outputs,
    # title=title,
    # examples=examples,
    # description=description,
    # cache_examples=True,
    theme='huggingface'
)
interface.launch(debug=True, enable_queue=True)