File size: 528 Bytes
8fbdf99
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
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()