Ant-Onion commited on
Commit
51c84a8
·
1 Parent(s): d214ea8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -1
app.py CHANGED
@@ -1,10 +1,18 @@
1
  import gradio as gr
2
  from pathlib import Path
3
  from transformers import pipeline
 
 
4
 
5
  pipeline = pipeline("image-classification", model="Ant-Onion/dummy-model", feature_extractor = "Ant-Onion/dummy-model")
6
 
7
- examples = "./images"
 
 
 
 
 
 
8
 
9
  def predict(image):
10
  predictions = pipeline(image)
 
1
  import gradio as gr
2
  from pathlib import Path
3
  from transformers import pipeline
4
+ from PIL import Image
5
+ import os, os.path
6
 
7
  pipeline = pipeline("image-classification", model="Ant-Onion/dummy-model", feature_extractor = "Ant-Onion/dummy-model")
8
 
9
+
10
+
11
+ imgs = []
12
+ path = "./images"
13
+
14
+ lista = os.listdir(path)
15
+ examples =list(map(lambda x: os.path.join(path,x), lista))
16
 
17
  def predict(image):
18
  predictions = pipeline(image)