Update app.py

app.py

@@ -12,20 +12,12 @@ class QModel():
         if (n_questions==1):
             return np.argmax(self.qtable[state_n][:])
         return np.argsort(self.qtable[state_n][:])[-n_questions:]
-        
-    #returns n states
-    '''
-        def predict(self, state_n, n_questions):
-        state_n = int(state_n)
-        n_questions = int(n_questions)
-        return np.argsort(self.qtable[state_n][:])[-n_questions:]
-    '''    
 
 learner = pickle.load(open("q-learn-multiply-game.pkl", "rb"))
 
-def predict_next_questions(state):
+def predict_next_questions(state, n_questions):
     
-    return learner.predict(state,5)
+    return learner.predict(state, n_questions)
 
 
 intf = gr.Interface(fn=predict_next_questions, inputs="number", outputs="json")