Shoaib-33 commited on
Commit
a22280d
·
1 Parent(s): b996a90

nnn111111

Browse files
Files changed (1) hide show
  1. app.py +6 -16
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
- model = load_learner(f'latest.pkl')
 
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',