kanad commited on
Commit
808a188
·
1 Parent(s): 9fd2c61

Add app file

Browse files
Files changed (1) hide show
  1. app.py +15 -8
app.py CHANGED
@@ -4,15 +4,22 @@ import gradio as gr
4
 
5
  learn = load_learner('model.pkl')
6
 
7
- categories = ('wet','tawny','horned')
8
 
9
- def classify_image(img):
10
- pred,idx,probs = learn.predict(img)
11
- return dict(zip(categories, map(float,probs)))
 
 
 
 
 
 
 
 
 
 
 
12
 
13
- image = gr.inputs.Image(shape=(224,224))
14
- label = gr.outputs.label()
15
- examples = ['harpy.jpg','horned.jpg']
16
 
17
- iface = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)
18
  iface.launch()
 
4
 
5
  learn = load_learner('model.pkl')
6
 
7
+ #categories = ('wet','tawny','horned')
8
 
9
+ #def classify_image(img):
10
+ # pred,idx,probs = learn.predict(img)
11
+ # return dict(zip(categories, map(float,probs)))
12
+
13
+ #image = gr.inputs.Image(shape=(224,224))
14
+ #label = gr.outputs.label()
15
+ #examples = ['harpy.jpg','horned.jpg']
16
+
17
+ #iface = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)
18
+
19
+ def greet(name):
20
+ return "Hello " + name + "!"
21
+
22
+ iface = gr.Interface(fn=greet, inputs="text", outputs="text")
23
 
 
 
 
24
 
 
25
  iface.launch()