chador2003 commited on
Commit
5738df6
·
verified ·
1 Parent(s): cbb2e86
Files changed (1) hide show
  1. app.py +10 -15
app.py CHANGED
@@ -1,18 +1,13 @@
1
- import gradio as gr
2
  from transformers import pipeline
 
3
 
4
- pipeline = pipeline(task="image-classification", model="julien-c/hotdog-not-hotdog")
5
-
6
- def predict(input_img):
7
- predictions = pipeline(input_img)
8
- return input_img, (p["label"]: p["score"] for p in predictions)
 
9
 
10
- gradio_app = gr.Interface(
11
- predict,
12
- inputs=gr.Image(label="Select hot dog candidate", sources=["upload", "webcam"], type="pil"),
13
- outputs=[gr.Image(label="Processed Image"), gr.Label(label="Result", num_top_classes=2)],
14
- title="Hot Dog? Or Not?",
15
- )
16
-
17
- if __name__ == "__main__":
18
- gradio_app.launch()
 
 
1
  from transformers import pipeline
2
+ classifier = pipeline('image-classification', model = "computer_partsclassifier-model")import numpy as np
3
 
4
+ def image_classifier(inp):
5
+ confidence_scores = np.random.rand(3)
6
+ confidence_scores/= np.sum(confidence_scores)
7
+ classes= ['cable','case', 'cpu']
8
+ result= {classes[i]: confidence_scores[i] for i in range(3)}
9
+ return result
10
 
11
+ import gradio as gr
12
+ demo = gr.Interface(fn=image_classifier, inputs = "image", outputs = "label")
13
+ demo.launch()