AlexN commited on
Commit
e58f728
1 Parent(s): 3b48807
__pycache__/TractionModel.cpython-37.pyc ADDED
Binary file (1.79 kB). View file
 
app.py CHANGED
@@ -33,8 +33,12 @@ else:
33
  device = torch.device("cpu")
34
  model = model.to(device)
35
 
36
- iface = gr.Interface(inference, live=True, inputs=gr.inputs.Image(source="upload", tool=None, type='pil'),
37
- outputs=gr.outputs.Label())
 
 
 
 
38
 
39
  iface.test_launch()
40
  if __name__ == "__main__":
 
33
  device = torch.device("cpu")
34
  model = model.to(device)
35
 
36
+
37
+ examples = [['tibo.png'], ['tibo2.png'], ['real_pull_up.png'], ['no_pull_up.png'], ['doge.jpg']]
38
+ iface = gr.Interface(inference, live=True, inputs=gr.inputs.Image(source="upload", type='pil'),
39
+ outputs=gr.outputs.Label(),
40
+ examples=examples,
41
+ enable_queue=True)
42
 
43
  iface.test_launch()
44
  if __name__ == "__main__":
doge.jpg ADDED
gradio_queue.db ADDED
Binary file (963 kB). View file
 
no_pull_up.png ADDED
real_pull_up.png ADDED
tibo.png ADDED
tibo2.png ADDED