TH9817 commited on
Commit
36a91f0
1 Parent(s): 265cc65

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -41,10 +41,10 @@ def read_video_pyav(container, indices):
41
  frames.append(frame)
42
  return np.stack([x.to_ndarray(format="rgb24") for x in frames])
43
 
44
- def chat(token):
45
  # Download video from the hub
46
  #video_path_1 = hf_hub_download(repo_id="raushan-testing-hf/videos-test", filename="sample_demo_1.mp4", repo_type="dataset")
47
- video_path="./sample1-Scene-001.mp4"
48
  #video_path_2 = hf_hub_download(repo_id="raushan-testing-hf/videos-test", filename="karate.mp4", repo_type="dataset")
49
 
50
  container = av.open(video_path)
@@ -98,7 +98,7 @@ def chat(token):
98
 
99
  demo = gr.Interface(
100
  fn=chat,
101
- inputs=[gr.Slider(100,300)],
102
  outputs=["text"],
103
  )
104
 
 
41
  frames.append(frame)
42
  return np.stack([x.to_ndarray(format="rgb24") for x in frames])
43
 
44
+ def chat(video_number,token):
45
  # Download video from the hub
46
  #video_path_1 = hf_hub_download(repo_id="raushan-testing-hf/videos-test", filename="sample_demo_1.mp4", repo_type="dataset")
47
+ video_path="./sample1-Scene-{0}.mp4".format(video_number)
48
  #video_path_2 = hf_hub_download(repo_id="raushan-testing-hf/videos-test", filename="karate.mp4", repo_type="dataset")
49
 
50
  container = av.open(video_path)
 
98
 
99
  demo = gr.Interface(
100
  fn=chat,
101
+ inputs=["text",gr.Slider(100,300)],
102
  outputs=["text"],
103
  )
104