davanstrien HF Staff commited on
Commit
4681ac2
·
1 Parent(s): 46ae1da

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -1
app.py CHANGED
@@ -1,3 +1,18 @@
1
  import gradio as gr
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
2
 
3
- gr.Interface.load("models/davanstrien/autotrain-encyclopaedia-illustrations-blog-post-3327992159").launch()
 
1
  import gradio as gr
2
+ from transformers import pipeline
3
+ from pathlib import Path
4
+
5
+ pipe = pipeline('image-classification','models/davanstrien/autotrain-encyclopaedia-illustrations-blog-post-3327992159')
6
+
7
+ def upload_file(files):
8
+ file_paths = [file.name for file in files]
9
+ return file_paths
10
+
11
+ with gr.Blocks() as demo:
12
+ file_output = gr.File()
13
+ upload_button = gr.UploadButton("Click to Upload a File", file_types=["image", "video"], file_count="multiple")
14
+ upload_button.upload(upload_file, upload_button, file_output)
15
+ gr.TextBox(len(list(Path(file_output))))
16
+
17
+ demo.launch()
18