porestar commited on
Commit
1195123
·
1 Parent(s): 0ed1317

Update app.py, score_db.py, rating.py, and requirements.txt

Browse files
Files changed (1) hide show
  1. score_db.py +4 -2
score_db.py CHANGED
@@ -28,9 +28,11 @@ class Winner(str, Enum):
28
  model_b = "model_b"
29
  tie = "tie"
30
 
 
31
  class Model(str, Enum):
32
- porestar_deepfault_unet_baseline_1 = "porestar/deepfault-unet-baseline-1"
33
- porestar_deepfault_unet_baseline_2 = "porestar/deepfault-unet-baseline-2"
 
34
 
35
  class Battle(BaseModel):
36
  model_a: Model
 
28
  model_b = "model_b"
29
  tie = "tie"
30
 
31
+
32
  class Model(str, Enum):
33
+ porestar_deepfault_unet_baseline_no_augment = "porestar/deepfault-unet-baseline-no-augment"
34
+ porestar_deepfault_unet_baseline_weak_augment = "porestar/deepfault-unet-baseline-weak-augment"
35
+ porestar_deepfault_unet_baseline_strong_augment = "porestar/deepfault-unet-baseline-strong-augment"
36
 
37
  class Battle(BaseModel):
38
  model_a: Model