aliabd HF staff commited on
Commit
2f89dbe
1 Parent(s): 993533e

Upload with huggingface_hub

Browse files
Files changed (5) hide show
  1. README.md +5 -5
  2. app.py +32 -0
  3. cheetah1.jpeg +0 -0
  4. cheetah1.jpg +0 -0
  5. lion.jpg +0 -0
README.md CHANGED
@@ -1,12 +1,12 @@
 
1
  ---
2
- title: Image Classifier Interface Load
3
- emoji: 😻
4
  colorFrom: indigo
5
- colorTo: green
6
  sdk: gradio
7
  sdk_version: 3.3.1
 
8
  app_file: app.py
9
  pinned: false
10
  ---
11
-
12
- Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
 
1
+
2
  ---
3
+ title: image_classifier_interface_load
4
+ emoji: 🔥
5
  colorFrom: indigo
6
+ colorTo: indigo
7
  sdk: gradio
8
  sdk_version: 3.3.1
9
+
10
  app_file: app.py
11
  pinned: false
12
  ---
 
 
app.py ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ import pathlib
3
+
4
+ current_dir = pathlib.Path(__file__)
5
+
6
+ images = [current_dir / "cheetah1.jpeg", current_dir / "cheetah1.jpg", current_dir / "lion.jpg"]
7
+
8
+
9
+ img_classifier = gr.Interface.load(
10
+ "models/google/vit-base-patch16-224", examples=images, cache_examples=True
11
+ )
12
+
13
+
14
+ def func(img, text):
15
+ return img_classifier(img), text
16
+
17
+
18
+ using_img_classifier_as_function = gr.Interface(
19
+ func,
20
+ [gr.Image(type="filepath"), "text"],
21
+ ["label", "text"],
22
+ examples=[
23
+ [current_dir / "cheetah1.jpeg", None],
24
+ [current_dir / "cheetah1.jpg", "cheetah"],
25
+ [current_dir / "lion.jpg", "lion"],
26
+ ],
27
+ cache_examples=True,
28
+ )
29
+ demo = gr.TabbedInterface([using_img_classifier_as_function, img_classifier])
30
+
31
+ if __name__ == "__main__":
32
+ demo.launch()
cheetah1.jpeg ADDED
cheetah1.jpg ADDED
lion.jpg ADDED