Update app.py

app.py

@@ -37,13 +37,17 @@ def process_result_detection_tab(text):
     '''
     mapping = {'llm': 'Machine Generated', 'human':'Human Written', 'machine-humanized': 'Machine Written, Machine Humanized', 'machine-polished': 'Human Written, Machine Polished'} 
 
-    result = classifier_abstract(text)[0]
-    result_r = classifier_essay(text)[0]
+    # Initialize scores for all classes
+    final_results = {label: 0.0 for label in mapping.values()}
 
-    labels = [mapping[x['label']] for x in result]
-    scores = list(0.5 * np.array([x['score'] for x in result]) + 0.5 * np.array([x['score'] for x in result_r]))
+    # Add scores from classifier_abstract
+    if result['label'] in mapping:
+        final_results[mapping[result['label']]] += 0.5 * result['score']
+
+    # Add scores from classifier_essay
+    if result_r['label'] in mapping:
+        final_results[mapping[result_r['label']]] += 0.5 * result_r['score']
 
-    final_results = dict(zip(labels, scores))
     print(final_results)
     return final_results