Kapu13 commited on
Commit
20abfd4
1 Parent(s): 59b1cb3

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +19 -0
app.py ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from transformers import pipeline
2
+
3
+ classifier = pipeline("image-classification", model="KayDee03/Fruits-model")
4
+
5
+ import numpy as np
6
+
7
+ # Function to classify images into 4 classes
8
+ def image_classifier(inp):
9
+ confidence_scores = np.random. rand(4)
10
+ confidence_scores /= np.sum(confidence_scores)
11
+ classes = ['Avocado', 'Banana', 'Guava', 'Mango']
12
+ result = {classes[i]: confidence_scores[i] for i in range(4)}
13
+ return result
14
+
15
+ import gradio as gr
16
+ # Creating Gradio interface
17
+ demo = gr.Interface(fn=image_classifier, inputs="image", outputs="label")
18
+ demo. launch()
19
+