IbrahimHasani commited on
Commit
3f1f1a2
1 Parent(s): c676d71

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -10,7 +10,7 @@ import tempfile
10
  import os
11
 
12
  # Load models
13
- resnet = ResNetModel.from_pretrained("Microsoft/resnet-50")
14
  resnet.eval()
15
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
16
  resnet = resnet.to(device)
@@ -143,8 +143,8 @@ iface = gr.Interface(
143
  inputs=[
144
  gr.Video(label="Upload a Video"),
145
  gr.Image(type="pil", label="Upload a Query Image"),
146
- gr.Slider(minimum=0, maximum=10, step=1, default=5, label="Skip Frames"),
147
- gr.Slider(minimum=0.0, maximum=1.0, step=0.01, default=0.47, label="Threshold")
148
  ],
149
  outputs=[
150
  gr.JSON(label="Result")
@@ -162,4 +162,4 @@ iface = gr.Interface(
162
  )
163
 
164
  if __name__ == "__main__":
165
- iface.launch()
 
10
  import os
11
 
12
  # Load models
13
+ resnet = ResNetModel.from_pretrained("microsoft/resnet-50")
14
  resnet.eval()
15
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
16
  resnet = resnet.to(device)
 
143
  inputs=[
144
  gr.Video(label="Upload a Video"),
145
  gr.Image(type="pil", label="Upload a Query Image"),
146
+ gr.Slider(minimum=0, maximum=10, step=1, value=5, label="Skip Frames"),
147
+ gr.Slider(minimum=0.0, maximum=1.0, step=0.01, value=0.47, label="Threshold")
148
  ],
149
  outputs=[
150
  gr.JSON(label="Result")
 
162
  )
163
 
164
  if __name__ == "__main__":
165
+ iface.launch()