Spaces:
Build error
Build error
nnn111111
Browse files
app.py
CHANGED
|
@@ -1,31 +1,21 @@
|
|
| 1 |
-
|
| 2 |
from fastai.vision.all import *
|
| 3 |
-
from fastai.vision.all import load_learner
|
| 4 |
import gradio as gr
|
| 5 |
|
| 6 |
-
|
| 7 |
-
|
| 8 |
-
|
| 9 |
-
"""import pathlib
|
| 10 |
-
plt = platform.system()
|
| 11 |
-
if plt == 'Windows': pathlib.WindowsPath = pathlib.PosixPath """
|
| 12 |
-
|
| 13 |
-
"""import pathlib
|
| 14 |
-
temp=pathlib.PosixPath
|
| 15 |
-
pathlib.PosixPath=pathlib.WindowsPath"""
|
| 16 |
|
| 17 |
cap_labels = ['AIDC F-CK-1 Ching-kuo', 'Boeing F-15EX Eagle II', 'Chengdu J-10 (China)', 'Chengdu J-20 (China)', 'Dassault Rafale', 'English Electric Lightning', 'Eurofighter Typhoon', 'Focke-Wulf Fw 190', 'General Dynamics F-16 Fighting Falcon aircraft', 'Grumman F-14 Tomcat', 'KAI KF-21 Boramae', 'Lockheed Martin F-22 Raptor', 'Lockheed Martin F-35 Lightning II', 'Lockheed P-80 Shooting Star', 'Lockheed YF-12', 'McDonnell Douglas F-4 Phantom II', 'Mikoyan MiG-29', 'Mikoyan Mig-31', 'Mikoyan-Gurevich MiG-25', 'Shenyang FC-31 Gyrfalcon', 'Sukhoi Su-27', 'Sukhoi Su-35 (Russia)', 'Sukhoi Su-57']
|
| 18 |
|
| 19 |
-
|
|
|
|
| 20 |
|
| 21 |
def recognize_image(image):
|
| 22 |
pred, idx, probs = model.predict(image)
|
| 23 |
return dict(zip(cap_labels, map(float, probs)))
|
| 24 |
|
| 25 |
-
|
| 26 |
-
#!export
|
| 27 |
image = gr.inputs.Image(shape=(192,192))
|
| 28 |
-
label = gr.outputs.Label()
|
| 29 |
examples = [
|
| 30 |
'images/euro.jpg',
|
| 31 |
'images/F-16.jpg',
|
|
|
|
|
|
|
| 1 |
from fastai.vision.all import *
|
|
|
|
| 2 |
import gradio as gr
|
| 3 |
|
| 4 |
+
# import pathlib
|
| 5 |
+
# temp = pathlib.PosixPath
|
| 6 |
+
# pathlib.PosixPath = pathlib.WindowsPath
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 7 |
|
| 8 |
cap_labels = ['AIDC F-CK-1 Ching-kuo', 'Boeing F-15EX Eagle II', 'Chengdu J-10 (China)', 'Chengdu J-20 (China)', 'Dassault Rafale', 'English Electric Lightning', 'Eurofighter Typhoon', 'Focke-Wulf Fw 190', 'General Dynamics F-16 Fighting Falcon aircraft', 'Grumman F-14 Tomcat', 'KAI KF-21 Boramae', 'Lockheed Martin F-22 Raptor', 'Lockheed Martin F-35 Lightning II', 'Lockheed P-80 Shooting Star', 'Lockheed YF-12', 'McDonnell Douglas F-4 Phantom II', 'Mikoyan MiG-29', 'Mikoyan Mig-31', 'Mikoyan-Gurevich MiG-25', 'Shenyang FC-31 Gyrfalcon', 'Sukhoi Su-27', 'Sukhoi Su-35 (Russia)', 'Sukhoi Su-57']
|
| 9 |
|
| 10 |
+
|
| 11 |
+
model = load_learner('latest.pkl')
|
| 12 |
|
| 13 |
def recognize_image(image):
|
| 14 |
pred, idx, probs = model.predict(image)
|
| 15 |
return dict(zip(cap_labels, map(float, probs)))
|
| 16 |
|
|
|
|
|
|
|
| 17 |
image = gr.inputs.Image(shape=(192,192))
|
| 18 |
+
label = gr.outputs.Label(num_top_classes=5)
|
| 19 |
examples = [
|
| 20 |
'images/euro.jpg',
|
| 21 |
'images/F-16.jpg',
|