CLIP / app.py
sudhir2016's picture
Update app.py
5883283 verified
raw
history blame
561 Bytes
import gradio as gr
from transformers import pipeline
pipe = pipeline("zero-shot-image-classification", model="openai/clip-vit-base-patch32")
def predict(photo,Label1,Label2,Label3):
out1= pipe(images=photo,candidate_labels=[Label1,Label2,Label3])
out2=out1[0]
out3=out2['label']
out4=out2['score']
if out4<0.9:
out3='No Match'
return out3
label1='Apple'
label2='Orange'
label3='Banana'
demo = gr.Interface(fn=predict,inputs=[gr.Image(type="pil"),'text','text','text'],outputs='text',examples=[['Banana.jpg',label1,label2,label3]])
demo.launch()