PDG commited on
Commit
612a23d
·
1 Parent(s): 6b2d891

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -57,10 +57,10 @@ def classifyCar(im):
57
  scores = torch.nn.functional.softmax(model(im)[0])
58
  return {LABELS[i]: float(scores[i]) for i in range(2)}
59
 
60
- examples = [[example_img.jpg], [example_img2.jpg]] # must be uploaded in repo
61
 
62
  # create interface for model
63
- interface = gr.Interface(classifyCar, inputs='Image', outputs='label', cache_examples=False, title='VW Up or Fiat 500', example=examples)
64
  interface.launch()
65
 
66
 
 
57
  scores = torch.nn.functional.softmax(model(im)[0])
58
  return {LABELS[i]: float(scores[i]) for i in range(2)}
59
 
60
+ #examples = [[example_img.jpg], [example_img2.jpg]] # must be uploaded in repo
61
 
62
  # create interface for model
63
+ interface = gr.Interface(classifyCar, inputs='Image', outputs='label', cache_examples=False, title='VW Up or Fiat 500')
64
  interface.launch()
65
 
66