Upload app correcao

app.py

@@ -5,15 +5,32 @@ Automatically generated by Colaboratory.
 
 Original file is located at
     https://colab.research.google.com/drive/1I-y23tTTs_aMjSYwPa3Fk8Ywz8H8ZUgl
+
+#**GRADIO**
 """
 
+!pip install -q gradio
+
+!pip install joblib
+
 import gradio as gr
 import joblib as jb
 
 def predict(sex, age, hypertension, heart_disease, ever_married, work_type, Residence_type, avg_glucose_level, bmi, smoking_status):
     model = jb.load('model.pkl')
+    sex = str(sex)
+    age = int(age)
+    hypertension = str(hypertension)
+    heart_disease = str(heart_disease)
+    ever_married = str(ever_married)
+    work_type = str(work_type)
+    Residence_type = int(Residence_type)
+    avg_glucose_level   = float(avg_glucose_level  )
+    bmi = float(bmi)
+    smoking_status = str(smoking_status)
 
-    p = model.predict_proba({{sex, age, hypertension, heart_disease, ever_married, work_type, Residence_type, avg_glucose_level, bmi, smoking_status}})[0]
+
+    p = model.predict({{sex, age, hypertension, heart_disease, ever_married, work_type, Residence_type, avg_glucose_level, bmi, smoking_status}})[0]
 
     return {"Não terá AVC": p[0], "Terá AVC": p[1]}
 
@@ -29,4 +46,7 @@ demo = gr.Interface(fn=predict,
                            "number",
                            gr.Dropdown(choices=["Nunca fumou", "Fuma"], type="index"),
                           ],
-                    outputs="label")
+                    outputs="label")
+
+
+demo.launch(share=True)