aliabd HF staff commited on
Commit
c2aceba
1 Parent(s): b1c4b56

Upload with huggingface_hub

Browse files
Files changed (2) hide show
  1. README.md +1 -1
  2. run.py +32 -0
README.md CHANGED
@@ -6,6 +6,6 @@ colorFrom: indigo
6
  colorTo: indigo
7
  sdk: gradio
8
  sdk_version: 3.4.1
9
- app_file: app.py
10
  pinned: false
11
  ---
 
6
  colorTo: indigo
7
  sdk: gradio
8
  sdk_version: 3.4.1
9
+ app_file: run.py
10
  pinned: false
11
  ---
run.py ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ import pathlib
3
+
4
+ current_dir = pathlib.Path(__file__).parent
5
+
6
+ images = [str(current_dir / "cheetah1.jpeg"), str(current_dir / "cheetah1.jpg"), str(current_dir / "lion.jpg")]
7
+
8
+
9
+ img_classifier = gr.Interface.load(
10
+ "models/google/vit-base-patch16-224", examples=images, cache_examples=False
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
+ [str(current_dir / "cheetah1.jpeg"), None],
24
+ [str(current_dir / "cheetah1.jpg"), "cheetah"],
25
+ [str(current_dir / "lion.jpg"), "lion"],
26
+ ],
27
+ cache_examples=False,
28
+ )
29
+ demo = gr.TabbedInterface([using_img_classifier_as_function, img_classifier])
30
+
31
+ if __name__ == "__main__":
32
+ demo.launch()