Update app.py

app.py

@@ -8,8 +8,8 @@ class QModel():
         
     #returns the top n states
     def predict(self,state_n):
-        print(self.qtable[int(state_n)])
-        np.argmax(self.qtable[int(state_n)])
+        state_n = int(state_n)
+        np.argmax(self.qtable[state_n][:])
 
 learner = pickle.load(open("q-learn-multiply-game.pkl", "rb"))