svalcin commited on
Commit
7cb45aa
1 Parent(s): 608b829

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -23,12 +23,12 @@ class QModel():
23
 
24
  learner = pickle.load(open("q-learn-multiply-game.pkl", "rb"))
25
 
26
- def predict_next_questions(state, n=1):
27
 
28
- return learner.predict(state,n)
29
 
30
 
31
- intf = gr.Interface(fn=predict_next_questions, inputs="number", outputs="number")
32
 
33
  intf.launch()
34
 
 
23
 
24
  learner = pickle.load(open("q-learn-multiply-game.pkl", "rb"))
25
 
26
+ def predict_next_questions(state):
27
 
28
+ return learner.predict(state,5)
29
 
30
 
31
+ intf = gr.Interface(fn=predict_next_questions, inputs="number", outputs=["number","number", "number", "number", "number"])
32
 
33
  intf.launch()
34