File size: 419 Bytes
78bf7d6
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
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()