AjaykumarPilla commited on
Commit
f048faf
·
verified ·
1 Parent(s): 2f7e2e5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -4
app.py CHANGED
@@ -161,9 +161,11 @@ def create_pitch_map(pitch_x, pitch_y):
161
 
162
  # Main Gradio function
163
  def drs_analysis(video):
164
- video_path = "temp_video.mp4"
165
- with open(video_path, "wb") as f:
166
- f.write(video.read())
 
 
167
 
168
  positions, frame_numbers, bounce_point, frame_rate, frame_width, frame_height = process_video(video_path)
169
  if not positions:
@@ -182,7 +184,8 @@ def drs_analysis(video):
182
 
183
  fig_pitch = create_pitch_map(pitch_x, pitch_y)
184
 
185
- os.remove(video_path)
 
186
 
187
  return fig_traj, fig_pitch, f"LBW Decision: {lbw_decision}\nSpeed: {speed_kmh:.2f} km/h", video_path
188
 
 
161
 
162
  # Main Gradio function
163
  def drs_analysis(video):
164
+ # Video is a file path (string) in Hugging Face Spaces
165
+ video_path = video if isinstance(video, str) else "temp_video.mp4"
166
+ if not isinstance(video, str):
167
+ with open(video_path, "wb") as f:
168
+ f.write(video.read())
169
 
170
  positions, frame_numbers, bounce_point, frame_rate, frame_width, frame_height = process_video(video_path)
171
  if not positions:
 
184
 
185
  fig_pitch = create_pitch_map(pitch_x, pitch_y)
186
 
187
+ if not isinstance(video, str):
188
+ os.remove(video_path)
189
 
190
  return fig_traj, fig_pitch, f"LBW Decision: {lbw_decision}\nSpeed: {speed_kmh:.2f} km/h", video_path
191