Spaces:
Runtime error
Runtime error
# 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) |