| import gradio as gr | |
| import numpy as np | |
| # Function to classify images into 7 classes | |
| def image_classifier(inp): | |
| confidence_scores = np.random.rand(7) | |
| confidence_scores /= np.sum(confidence_scores) | |
| classes = ['Hor', 'Jadrima', 'Kishuthara', 'Marthra', 'Pangtse', 'Serthra', 'Shinglo'] | |
| result = {classes[i]: confidence_scores[i] for i in range(7)} | |
| return result | |
| # Creating Gradio interface | |
| demo = gr.Interface(fn=image_classifier, inputs="image", outputs="label") | |
| if __name__ == "__main__": | |
| demo.launch() |