Update app.py

app.py

@@ -23,11 +23,12 @@ class QModel():
 
 learner = pickle.load(open("q-learn-multiply-game.pkl", "rb"))
 
-def predict_next_questions(state, n):
+def predict_next_questions(state, n=1):
+    
     return learner.predict(state,n)
 
 
-intf = gr.Interface(fn=predict_next_questions, inputs=["number","number"], outputs=["number","number"])
+intf = gr.Interface(fn=predict_next_questions, inputs="number", outputs="number")
 
 intf.launch()