Dimasnoufal's picture
Add application file
78bf7d6 verified
raw
history blame contribute delete
419 Bytes
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()