QingJoy commited on
Commit
3cfe126
·
1 Parent(s): 6fd6557

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +23 -0
app.py ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ from fastcore.all import *
3
+ from fastai.vision.all import *
4
+ import numpy as np
5
+ import gradio as gr
6
+ import cv2
7
+
8
+ def isbird(image):
9
+ #加载模型
10
+ learn = load_learner('./export.pkl')
11
+ #预测下一张图片
12
+ is_bird,_,probs = learn.predict(image)
13
+ print(f"This is a: {is_bird}.")
14
+ is_bird =str(f"This is a: {is_bird}.")
15
+ print(f"Probability it's a bird: {probs[0]:.4f}")
16
+ probability=str(f"Probability it's a bird: {probs[0]:.4f}")
17
+ return is_bird,probability
18
+
19
+ demo = gr.Interface(fn=isbird,
20
+ inputs="image",
21
+ outputs="text",
22
+ title="Is this a bird?",)
23
+ demo.launch()