Spaces:
Sleeping
Sleeping
File size: 595 Bytes
1891bb2 f41f480 1891bb2 f41f480 1891bb2 f41f480 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
from transformers import pipeline
classifier = pipeline("image-classification", model="Dalaix703/flowerr-model")
import gradio as gr
import numpy as np
# Function to classify images into 7 classes
def image_classifier(inp):
confidence_scores = np.random.rand(5)
confidence_scores /= np.sum(confidence_scores)
classes = ['crocus', 'daffodil', 'daisy', 'dandelion', 'fritillary']
result = {classes[i]: confidence_scores[i] for i in range(5)}
return result
# Creating Gradio interface
demo = gr.Interface(fn=image_classifier, inputs="image", outputs="label")
demo.launch(share=True) |