Upload app.py

app.py

@@ -15,10 +15,10 @@ metric_scale = {
     'clip_score': 0.02,
     'env_clip': 1.5,
     'other_subject_clip': 1.5,
-    'image_quality': 1,
-    'dynamic_degree': 1,
-    'aesthetic_quality': 1,
-    'motion_smoothness': 1,
+    'image_quality': 1.0,
+    'dynamic_degree': lambda x: 0.1 if x < 0.885 else (0.95 if x < 0.95 and x > 0.88 else 1.0),
+    'aesthetic_quality': 1.0,
+    'motion_smoothness': 1.0,
 }
 
 class ModelResult:
@@ -39,11 +39,16 @@ class ModelResult:
         total_score = []
         for metric in self.result.keys():
             res[metric] = round(float(self.result[metric]) - 1e-3, 4)
-            total_score.append(self.result[metric] * metric_scale[metric])
+            if metric == 'dynamic_degree':
+                score = metric_scale[metric](self.result[metric]) * self.result[metric]
+            else:
+                score = self.result[metric] * metric_scale[metric]
+            total_score.append(score)
         total_score = sum(total_score) / len(total_score)
         res['comprehensive score'] = round(total_score, 4)
         return res
 
+
     
 def eval_request(model_name, org_link, huggingface_data_set_name):
     params = {