before sleep

README.md

@@ -8,7 +8,7 @@ sdk_version: 6.5.1
 app_file: app.py
 pinned: false
 license: mit
-short_description: simple app for the purpose of novelty challange
+short_description: simple app for the purpose of novelty challenge
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -114,12 +114,15 @@ def process_id(id_number, session_id):
             ref_embeddings = result
 
     yield "Calculating the final score...", api_data_display, None, None
-    score = calculate_score(paper_embedding, ref_embeddings)
+    score1 = calculate_score(paper_embedding, ref_embeddings)
+    score2 = calculate_score(None, ref_embeddings)
+    score = (score1 * score2) / (score1 + score2)
+    normalized_score = (score - 0.01) / (0.1 - 0.01) # todo adjust min/max based on real data
     time.sleep(0.5)
 
     result_message = f"🎉 Processing complete! Score calculated successfully."
     
-    yield result_message, api_data_display, score, None 
+    yield result_message, api_data_display, score, normalized_score 
 
 
 # Create Gradio interface

utils.py

@@ -80,6 +80,8 @@ def create_abstract(abstract_index):
 def calculate_score(paper_embedding, ref_embeddings):
     if ref_embeddings.shape[0] == 0:
         return 0.0
+    if paper_embedding is None:
+        paper_embedding = ref_embeddings.mean(axis=0).unsqueeze(0)
     similarities = torch.nn.functional.cosine_similarity(
         paper_embedding,
         ref_embeddings,