Update app.py

app.py

@@ -1,19 +1,18 @@
+import gradio as gr
 import joblib
 import pandas as pd
-import gradio as gr
 
 # Carregar o modelo
 model_LayAway = joblib.load('model_class_layaway_0.pkl')
 model_LayHome = joblib.load('model_class_layhome_1.pkl')
 
-# Função para fazer previsões com os modelos
+
 def fazer_previsao(ODD_H, ODD_D, ODD_A):
-    data = pd.DataFrame({'H/A': [ODD_H / ODD_A], 'Diff': [(ODD_H / ODD_D) - (ODD_H / ODD_A)]})
+    data = pd.DataFrame(
+        {'H/A': [ODD_H / ODD_A], 'Diff': [(ODD_H / ODD_D) - (ODD_H / ODD_A)]})
     previsao_LayAway = model_LayAway.predict(data)[0]
     previsao_LayHome = model_LayHome.predict(data)[0]
-    #previsao_H = model_LayAway.predict_proba(data)[1]
-    #previsao_A = model_LayHome.predict_proba(data)[1]
-    
+
     if (previsao_LayAway == previsao_LayHome):
         resultado = 'Não entre!'
     else:
@@ -22,13 +21,10 @@ def fazer_previsao(ODD_H, ODD_D, ODD_A):
         else:
             entrada_LayAway = 'NAO'
 
-        if previsao_LayHome == 1:
-            entrada_LayHome = 'SIM'
-        else:
-            entrada_LayHome = 'NAO'
-
-        probabilidade_LayAway = round(100 * model_LayAway.predict_proba(data)[0][1], 2)
-        probabilidade_LayHome = round(100 * model_LayHome.predict_proba(data)[0][1], 2)
+        probabilidade_LayAway = round(
+            100 * model_LayAway.predict_proba(data)[0][1], 2)
+        probabilidade_LayHome = round(
+            100 * model_LayHome.predict_proba(data)[0][1], 2)
 
         if probabilidade_LayAway > 68.2:
             entrada_Home = 'SIM'
@@ -39,20 +35,25 @@ def fazer_previsao(ODD_H, ODD_D, ODD_A):
             entrada_Away = 'SIM'
         else:
             entrada_Away = 'NAO'
-            
-            
+
+        if previsao_LayHome == 1:
+            entrada_LayHome = 'SIM'
+        else:
+            entrada_LayHome = 'NAO'
+
         resultado = f"entrada LayAway (se Odd> 1.17): {entrada_LayAway}\n" \
                     f"------ probabilidade: {probabilidade_LayAway}%\n" \
                     f"\n" \
                     f"entrada Home: {entrada_Home}\n" \
                     f"\n" \
                     f"entrada LayHome (se Odd> 1.28): {entrada_LayHome}\n" \
-                    f"------ probabilidade: {probabilidade_LayHome}%"
+                    f"------ probabilidade: {probabilidade_LayHome}%\n" \
                     f"\n" \
-                    f"entrada Home: {entrada_Away}\n" \
+                    f"entrada Away: {entrada_Away}"
 
     return resultado
 
+
 # Criar uma interface Gradio
 iface = gr.Interface(
     fn=fazer_previsao,
@@ -64,4 +65,4 @@ iface = gr.Interface(
     description="Insira as Odds e obtenha uma sugestão para entradas LayAway (CASA vence ou EMPATA) e LayHome (VISITANTE vence ou EMPATA)."
 )
 
-iface.launch()
+iface.launch()