init app

app.py

@@ -3,6 +3,7 @@ import random
 
 from ml.model import base_df, ml_model
 from ml.predictor import Predictor
+from ml.utils import load_pickle
 
 
 def function(team1, team2):
@@ -20,16 +21,20 @@ def function(team1, team2):
         }
     else:
         return {
-            'result': winner,
+            'winner': winner,
             'probability': winner_proba
         }
 
 
 predictor = Predictor(base_df, ml_model)
+examples = random.choices([x[1:3] for x in load_pickle("data/table_match.pkl")['matches']], k=20)
+examples = [list(x) for x in examples]
 iface = gr.Interface(fn=function,
                      inputs=[gr.Textbox(placeholder="Qatar"), gr.Textbox(placeholder="Ecuador")],
                      outputs="json",
-                     title="WorldCup-Prediction \n "
-                           "Predicting the 2022 FIFA World Cup results with Machine Learning!"
+                     title="WorldCup-Prediction \n\n "
+                           "Predicting the 2022 FIFA World Cup results with Machine Learning!",
+                     examples=examples
                      )
+iface.queue(concurrency_count=5)
 iface.launch()