ramonvictorn commited on
Commit
f96a346
1 Parent(s): fc07b9a

feat: addmin better examples and remove old cod

Browse files
Files changed (4) hide show
  1. app.py +7 -28
  2. audi_a5.jpeg +0 -0
  3. ferrari.jpg +0 -0
  4. kwid.png +0 -0
app.py CHANGED
@@ -2,11 +2,6 @@ import gradio as gr
2
  from fastai.vision.all import *
3
  from huggingface_hub import from_pretrained_fastai
4
 
5
-
6
- def is_imported(x): return x[o].isupper()
7
-
8
-
9
- # learn = load_learner("model.pkl")
10
  learn = from_pretrained_fastai("ramonvictorn/brazilian_car_check")
11
  categories = ('Imported', 'Nacional')
12
 
@@ -16,28 +11,12 @@ def classify_image(img):
16
  return dict(zip(categories, map(float, props)))
17
 
18
 
19
- image = gr.Image(shape=(192, 192), type="pil")
20
  label = gr.outputs.Label()
21
- # examples = ["xx.jpg"]
22
-
23
- gr.Interface(fn=classify_image,
24
- inputs=gr.Image(type="pil"),
25
- outputs=label,
26
- examples=["celta.jpeg"]).launch(inline=True)
27
-
28
-
29
- # intf = gr.Interface(fn=classify_image, inputs=image,
30
- # outputs=label, examples=["./celta.jpg"])
31
- # intf.launch(inline=True)
32
- # pipeline = pipeline(task="image-classification", model="julien-c/hotdog-not-hotdog")
33
-
34
- # def predict(image):
35
- # predictions = pipeline(image)
36
- # return {p["label"]: p["score"] for p in predictions}
37
 
38
- # gr.Interface(
39
- # predict,
40
- # inputs=gr.inputs.Image(label="Upload car candidate", type="filepath"),
41
- # outputs=gr.outputs.Label(num_top_classes=2),
42
- # title="Brazilian nacional? Or Imported?",
43
- # ).launch()
 
2
  from fastai.vision.all import *
3
  from huggingface_hub import from_pretrained_fastai
4
 
 
 
 
 
 
5
  learn = from_pretrained_fastai("ramonvictorn/brazilian_car_check")
6
  categories = ('Imported', 'Nacional')
7
 
 
11
  return dict(zip(categories, map(float, props)))
12
 
13
 
14
+ image = gr.Image(shape=(192, 192), type="pil", label="Upload a car candidate")
15
  label = gr.outputs.Label()
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
16
 
17
+ intf = gr.Interface(
18
+ fn=classify_image,
19
+ inputs=image,
20
+ outputs=label, examples=["ferrari.jpg","kwid.png", "celta.jpeg"]
21
+ )
22
+ intf.launch(inline=True)
audi_a5.jpeg ADDED
ferrari.jpg ADDED
kwid.png ADDED