surely is broken

app.py

@@ -51,6 +51,7 @@ def compute_itm_score(image, statement):
     score = itm_scores[:, 1].item()
     logging.info('Finished compute_itm_score')
     return score
+
 def generate_caption(processor, model, image):
     logging.info('Starting generate_caption')
     inputs = processor(images=image, return_tensors="pt").to(device)
@@ -72,18 +73,18 @@ def save_dataframe_to_csv(df):
     # Return the file path (no need to reopen the file with "rb" mode)
     return temp_file_path
 
-
-# Define a validation function to check if the uploaded file is an image
-def validate_image_file(file):
+# Define a function to check if the uploaded file is an image
+def is_image_file(file):
     allowed_extensions = [".jpg", ".jpeg", ".png", ".gif", ".bmp", ".tiff"]
     file_extension = os.path.splitext(file.name)[1]
-    if file_extension.lower() not in allowed_extensions:
-        return "Invalid file type. Please upload an image file (e.g., .jpg, .png, .jpeg)."
-    return None
+    return file_extension.lower() in allowed_extensions
+
 
 # Main function to perform image captioning and image-text matching
 def process_images_and_statements(file):
-    logging.info('Starting process_images_and_statements')
+ # Check if the uploaded file is an image
+    if not is_image_file(file):
+        return "Invalid file type. Please upload an image file (e.g., .jpg, .png, .jpeg)."
     
     # Extract the filename from the file object
     filename = file.name
@@ -135,7 +136,7 @@ def process_images_and_statements(file):
     return results_df, csv_results # <--- Return results_df and csv_results
 
 # Gradio interface
-file_input = gr.inputs.File(label="Upload Image", validate=validate_image_file)  # Use File input for image upload
+file_input = gr.inputs.File(label="Upload Image")  # Use File input for image upload
 output_df = gr.outputs.Dataframe(type="pandas", label="Results")
 output_csv = gr.outputs.File(label="Download CSV")
 
@@ -149,4 +150,4 @@ iface = gr.Interface(
 )
 
 # Launch the Gradio interface
-iface.launch()
+iface.launch()