capradeepgujaran commited on
Commit
5a74233
·
verified ·
1 Parent(s): 230a814

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -79,7 +79,7 @@ def analyze_construction_image(images=None, video=None):
79
 
80
  if images:
81
  for i, image_file in enumerate(images):
82
- image = Image.open(image_file.name)
83
  resized_image = resize_image(image) # Resize image before processing
84
  image_data_url = f"data:image/png;base64,{encode_image(resized_image)}"
85
  messages = [
@@ -112,7 +112,7 @@ def analyze_construction_image(images=None, video=None):
112
  results.append((f"Image {i+1} analysis", result))
113
 
114
  if video:
115
- frames = extract_frames_from_video(video.name) # Extract fewer frames from video
116
  for i, frame in enumerate(frames):
117
  image_data_url = f"data:image/png;base64,{encode_image(frame)}"
118
  messages = [
 
79
 
80
  if images:
81
  for i, image_file in enumerate(images):
82
+ image = Image.open(image_file.name) # For image uploads, we use image_file.name
83
  resized_image = resize_image(image) # Resize image before processing
84
  image_data_url = f"data:image/png;base64,{encode_image(resized_image)}"
85
  messages = [
 
112
  results.append((f"Image {i+1} analysis", result))
113
 
114
  if video:
115
+ frames = extract_frames_from_video(video) # Use video directly, as it's a file path
116
  for i, frame in enumerate(frames):
117
  image_data_url = f"data:image/png;base64,{encode_image(frame)}"
118
  messages = [