update theme, fix percentages for dataframe test display, update test statements

app.py

@@ -39,25 +39,25 @@ def compute_textual_similarity(caption, statement):
 
 # List of statements for Image-Text Matching
 statements = [
-    "contains a cartoon, figurine, or toy",
-    "is for children",
-    "children",
+    "contains or features a cartoon, figurine, or toy",
+    "appears to be for children",
+    "includes children",
     "sexual",
-    "depicts a child or portrays objects, images, or cartoon figures that primarily appeal to persons below the legal purchase age of alcohol",
-    "uses the name of or displays Santa Claus",
+    "nudity",
+    "depicts a child or portrays objects, images, or cartoon figures that primarily appeal to persons below the legal purchase age",
+    "uses the name of or depicts Santa Claus",
     'promotes alcohol use as a "rite of passage" to adulthood',
-    "uses brand identification—including logos, trademarks, or names—on clothing, toys, games, game equipment, or other items intended for use primarily by persons below the legal purchase age of alcohol",
+    "uses brand identification—including logos, trademarks, or names—on clothing, toys, games, game equipment, or other items intended for use primarily by persons below the legal purchase age",
     "portrays persons in a state of intoxication or in any way suggests that intoxication is socially acceptable conduct",
-    "makes curative or therapeutic claims",
+    "makes curative or therapeutic claims, except as permitted by law",
     "makes claims or representations that individuals can attain social, professional, educational, or athletic success or status due to beverage alcohol consumption",
     "degrades the image, form, or status of women, men, or of any ethnic group, minority, sexual orientation, religious affiliation, or other such group?",
     "uses lewd or indecent images or language",
-    "employs religion or religious themes",
+    "employs religion or religious themes?",
     "relies upon sexual prowess or sexual success as a selling point for the brand",
     "uses graphic or gratuitous nudity, overt sexual activity, promiscuity, or sexually lewd or indecent images or language",
     "associates with anti-social or dangerous behavior",
-    "depicts illegal activity of any kind?",
-    "illegal activity",
+    "depicts illegal activity",
     'uses the term "spring break" or sponsors events or activities that use the term "spring break," unless those events or activities are located at a licensed retail establishment',
 ]
 
@@ -104,13 +104,15 @@ def process_images_and_statements(image):
         itm_score_statement = compute_itm_score(image, statement)
 
         # Combine the two scores using a weighted average
-        final_score = (weight_textual_similarity * textual_similarity_score) + (weight_statement * itm_score_statement)
+        #final_score = (weight_textual_similarity * textual_similarity_score) + (weight_statement * itm_score_statement)
+        final_score = ((weight_textual_similarity * textual_similarity_score) +
+                       (weight_statement * itm_score_statement)) * 100  # Multiply by 100
 
         # Append the result to the DataFrame
         results_df = results_df.append({
             'Statement': statement,
-            'Textual Similarity Score': textual_similarity_score,
-            'ITM Score': itm_score_statement,
+            'Textual Similarity Score': textual_similarity_score * 100,  # Multiply by 100
+            'ITM Score': itm_score_statement * 100,  # Multiply by 100
             'Final Combined Score': final_score
         }, ignore_index=True)
 
@@ -128,6 +130,7 @@ iface = gr.Interface(
     inputs=image_input,
     outputs=output,
     title="Image Captioning and Image-Text Matching",
+    theme='sudeepshouche/minimalist',
     css=".output { flex-direction: column; } .output .outputs { width: 100%; }"  # Custom CSS
 )