Spaces:
Runtime error
Runtime error
import gradio as gr | |
from transformers import pipeline | |
from PIL import Image | |
def image_clasifier(input_image: gr.Image): | |
classifier1 = pipeline("image-classification", model="Dimasnoufal/image_strawbery-peach_classifier") | |
pil_image1 = Image.fromarray(input_image) | |
result1 = classifier1(pil_image1) | |
return result1 | |
iface = gr.Interface(fn=image_clasifier, inputs="image", outputs="json") | |
iface.launch() |