spark323 commited on
Commit
d61b319
·
1 Parent(s): 2936ffb

fix deprecation

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -16,6 +16,7 @@ transform = create_transform(
16
  )
17
  model.eval()
18
 
 
19
  def predict_fn(img):
20
  img = img.convert('RGB')
21
  img = transform(img).unsqueeze(0)
@@ -29,4 +30,5 @@ def predict_fn(img):
29
 
30
  return {LABELS[i]: v.item() for i, v in zip(indices, values)}
31
 
32
- gr.Interface(predict_fn, gr.inputs.Image(type='pil'), outputs='label').launch()
 
 
16
  )
17
  model.eval()
18
 
19
+
20
  def predict_fn(img):
21
  img = img.convert('RGB')
22
  img = transform(img).unsqueeze(0)
 
30
 
31
  return {LABELS[i]: v.item() for i, v in zip(indices, values)}
32
 
33
+
34
+ gr.Interface(predict_fn, gr.components.Image(type='pil'), outputs='label').launch()