aliabd commited on
Commit
f11f32f
·
1 Parent(s): 049276d

Upload with huggingface_hub

Browse files
Files changed (6) hide show
  1. README.md +6 -7
  2. cheetah1.jpeg +0 -0
  3. cheetah1.jpg +0 -0
  4. lion.jpg +0 -0
  5. requirements.txt +1 -0
  6. run.py +32 -0
README.md CHANGED
@@ -1,12 +1,11 @@
 
1
  ---
2
- title: Image Classifier Interface Load Main
3
- emoji: 💩
4
- colorFrom: gray
5
- colorTo: purple
6
  sdk: gradio
7
  sdk_version: 3.6
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_main
4
+ emoji: 🔥
5
+ colorFrom: indigo
6
+ colorTo: indigo
7
  sdk: gradio
8
  sdk_version: 3.6
9
+ app_file: run.py
10
  pinned: false
11
  ---
 
 
cheetah1.jpeg ADDED
cheetah1.jpg ADDED
lion.jpg ADDED
requirements.txt ADDED
@@ -0,0 +1 @@
 
 
1
+ https://gradio-main-build.s3.amazonaws.com/c3bec6153737855510542e8154391f328ac72606/gradio-3.6-py3-none-any.whl
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()