Update app.py

app.py

@@ -52,8 +52,8 @@ def construct_model_name(
     
     base += f"-{enc_dec_layers}l"
 
-    # if attention_heads_num == 1:
-    #     base += "-corrected"
+    if attention_heads_num == 1:
+        base += "-corrected"
     
     if d_model != 256:
         base += f"-{d_model}d"