taesiri commited on
Commit
13d5d73
1 Parent(s): 01c97d5
Files changed (1) hide show
  1. app.py +14 -3
app.py CHANGED
@@ -199,15 +199,26 @@ def row_selected_latest(evt: gr.SelectData):
199
  def load_video(url):
200
  post_id = get_reddit_id(url)
201
  video_url = f"https://huggingface.co/datasets/asgaardlab/GamePhysicsDailyDump/resolve/main/data/videos/{post_id}.mp4?download=true"
 
202
 
203
  # make sure file exists before returning, make a request without downloading the file
204
- r = requests.head(video_url)
205
- if r.status_code != 200 and r.status_code != 302:
 
 
 
 
 
 
 
206
  raise gr.Error(
207
  f"Video is not in the repo, please try another post. - {r.status_code }"
208
  )
209
 
210
- return video_url
 
 
 
211
 
212
 
213
  with gr.Blocks() as demo:
 
199
  def load_video(url):
200
  post_id = get_reddit_id(url)
201
  video_url = f"https://huggingface.co/datasets/asgaardlab/GamePhysicsDailyDump/resolve/main/data/videos/{post_id}.mp4?download=true"
202
+ video_url2 = f"https://huggingface.co/datasets/asgaardlab/GamePhysics-FullResolution/resolve/main/videos/{post_id}/{post_id}.mp4?download=true"
203
 
204
  # make sure file exists before returning, make a request without downloading the file
205
+ r1 = requests.head(video_url)
206
+ r2 = requests.head(video_url2)
207
+
208
+ if (
209
+ r2.status_code != 200
210
+ and r2.status_code != 302
211
+ and r1.status_code != 200
212
+ and r1.status_code != 302
213
+ ):
214
  raise gr.Error(
215
  f"Video is not in the repo, please try another post. - {r.status_code }"
216
  )
217
 
218
+ if r1.status_code == 200 or r1.status_code == 302:
219
+ return video_url
220
+ else:
221
+ return video_url2
222
 
223
 
224
  with gr.Blocks() as demo: