saakshigupta commited on
Commit
d1eeba1
·
verified ·
1 Parent(s): 2a3aaa9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -356,7 +356,7 @@ def process_image_with_xception_gradcam(image, model, device, pred_class):
356
  _, _, _, _, face_box, _ = dataset[0]
357
 
358
  return raw_cam, overlay, comparison, face_box
359
- else:
360
  st.error("Failed to generate GradCAM visualization")
361
  return None, None, None, None
362
 
@@ -435,18 +435,18 @@ def generate_gradcam_caption(image, processor, model, max_length=60):
435
  # Try to parse the caption based on different possible formats
436
  try:
437
  # Original format with "high activation:" etc.
438
- formatted_text = ""
439
  if "high activation :" in caption:
440
  high_match = caption.split("high activation :")[1].split("moderate")[0]
441
- formatted_text += f"**High activation**:\n{high_match.strip()}\n\n"
442
 
443
  if "moderate activation :" in caption:
444
  moderate_match = caption.split("moderate activation :")[1].split("low")[0]
445
- formatted_text += f"**Moderate activation**:\n{moderate_match.strip()}\n\n"
446
 
447
  if "low activation :" in caption:
448
  low_match = caption.split("low activation :")[1]
449
- formatted_text += f"**Low activation**:\n{low_match.strip()}"
450
 
451
  # If nothing was extracted using the original format, try alternative formats
452
  if not formatted_text.strip():
 
356
  _, _, _, _, face_box, _ = dataset[0]
357
 
358
  return raw_cam, overlay, comparison, face_box
359
+ else:
360
  st.error("Failed to generate GradCAM visualization")
361
  return None, None, None, None
362
 
 
435
  # Try to parse the caption based on different possible formats
436
  try:
437
  # Original format with "high activation:" etc.
438
+ formatted_text = ""
439
  if "high activation :" in caption:
440
  high_match = caption.split("high activation :")[1].split("moderate")[0]
441
+ formatted_text += f"**High activation**:\n{high_match.strip()}\n\n"
442
 
443
  if "moderate activation :" in caption:
444
  moderate_match = caption.split("moderate activation :")[1].split("low")[0]
445
+ formatted_text += f"**Moderate activation**:\n{moderate_match.strip()}\n\n"
446
 
447
  if "low activation :" in caption:
448
  low_match = caption.split("low activation :")[1]
449
+ formatted_text += f"**Low activation**:\n{low_match.strip()}"
450
 
451
  # If nothing was extracted using the original format, try alternative formats
452
  if not formatted_text.strip():