chador2003 commited on
Commit
f87a566
·
verified ·
1 Parent(s): 62d98f3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -1,5 +1,7 @@
1
  from transformers import pipeline
2
- classifier = pipeline('image-classification', model = "computerparts_classifier-model")
 
 
3
  import numpy as np
4
 
5
  def image_classifier(inp):
@@ -11,4 +13,4 @@ def image_classifier(inp):
11
 
12
  import gradio as gr
13
  demo = gr.Interface(fn=image_classifier, inputs = "image", outputs = "label")
14
- demo.launch()
 
1
  from transformers import pipeline
2
+ classifier = pipeline('image-classification', model = "computer_partsclassifier-model")
3
+ classifier(image)
4
+
5
  import numpy as np
6
 
7
  def image_classifier(inp):
 
13
 
14
  import gradio as gr
15
  demo = gr.Interface(fn=image_classifier, inputs = "image", outputs = "label")
16
+ demo.launch()