Update app.py

app.py

@@ -82,7 +82,8 @@ def stutterPrediction(f1, f2):
     preds = (net1(inputs).detach().cpu().numpy() + net2(inputs).detach().cpu().numpy()) / 2
 
   preds = sigmoid(preds)
-
+  names = np.loadtxt('classes.txt', dtype=str)
+  
   fig1 = plt.figure()
   for i in range(6,12):
     plt.plot(np.arange(preds.shape[0])*step/1000, preds[:,i], label=names[i])
@@ -91,8 +92,6 @@ def stutterPrediction(f1, f2):
   plt.ylabel("Class")
   plt.tight_layout()
 
-  names = np.loadtxt('classes.txt', dtype=str)
-
   dpreds = np.zeros(preds.shape)
 
   dpreds[0] = preds[0]