seamusbarnes commited on
Commit
d20620b
·
1 Parent(s): 38728e6
Files changed (2) hide show
  1. .gitattributes +2 -0
  2. app.py +6 -0
.gitattributes CHANGED
@@ -33,3 +33,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ *.jpg filter=lfs diff=lfs merge=lfs -text
37
+ samples/gothic[[:space:]]architecture[[:space:]]sample.jpg filter=lfs diff=lfs merge=lfs -text
app.py CHANGED
@@ -10,8 +10,14 @@ def predict(img):
10
  pred, pred_ind, probs = learn.predict(img)
11
  return {labels[i]: float(probs[i]) for i in range(len(probs))}
12
 
 
 
 
 
13
  gr.Interface(fn=predict,
14
  inputs='image',
15
  outputs='label',
 
 
16
  live=True
17
  ).launch()
 
10
  pred, pred_ind, probs = learn.predict(img)
11
  return {labels[i]: float(probs[i]) for i in range(len(probs))}
12
 
13
+ subfolder = Path('samples')
14
+ search_pattern = str(subfolder/'*.jpg')
15
+ jpg_files = glob.glob(search_pattern)
16
+
17
  gr.Interface(fn=predict,
18
  inputs='image',
19
  outputs='label',
20
+ examples=jpg_files,
21
+ examples_per_page=10,
22
  live=True
23
  ).launch()