Filter .model files out of model folder

ThatNerdUKnow · Sep 27, 2020 · 24763a3 · 24763a3
1 parent b0f71bb
commit 24763a3
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/vmaf-gui/Form1.cs b/vmaf-gui/Form1.cs
@@ -94,9 +94,19 @@ private void Form1_Load(object sender, EventArgs e)
             foreach (string model in models)
             {
                 string safeName = model.Substring(8, model.Length - 8);
-                cmbModel.Items.Add(safeName);
+                if (!safeName.Contains(".model")&& safeName.Contains(".pkl"))
+                {
+                    cmbModel.Items.Add(safeName);
+                }
+            }
+            try
+            {
+                cmbModel.SelectedIndex = 0;
+            }
+            catch
+            {
+                MessageBox.Show("There are no suitable models in the ./model folder");
             }
-            cmbModel.SelectedIndex = 0;
         }
 
         private void button1_Click(object sender, EventArgs e)