Update app.py

app.py

@@ -27,7 +27,7 @@ def wine(alcohol, chlorides, density, type, volatil_acidity,fixed_acidity, citri
     print("Predicting")
     print(df)
     
-    if type == "red":
+    if type == 0:
          res = model_red.predict(df) 
          print(res)
          wine_url = "https://raw.githubusercontent.com/Anniyuku/wine_quality/main/" + res[0] + ".png"
@@ -45,7 +45,7 @@ demo = gr.Interface(
     description="Experiment with type, alcohol, chlorides, density, volatil_acidity, fixed_acidity, citric_acid, total_sulfur_dioxide to predict which flower it is.",
     allow_flagging="never",
     inputs=[
-        gr.inputs.Radio(choices=['red','white'], label='Type'),
+        gr.inputs.Radio(choices=[0,1], label='Type'),
         gr.inputs.Number(default=9.00, label="alcohol"),
         gr.inputs.Number(default=0.60, label="chlorides"),
         gr.inputs.Number(default=1.00, label="density"),