Output Lable

app.py

@@ -38,9 +38,6 @@ def wine_quality(type, fixed_acidity, volatile_acidity, citric_acid, residual_su
     print(df)
     # 'res' is a list of predictions returned as the label.
     res = model.predict(df) 
-    # We add '[0]' to the result of the transformed 'res', because 'res' is a list, and we only want 
-    # the first element.
-#     print("Res: {0}").format(res)
     print(res)
         
 demo = gr.Interface(
@@ -62,7 +59,8 @@ demo = gr.Interface(
         gr.inputs.Number(default=5.0, label="sulphates"),
         gr.inputs.Number(default=5.0, label="alcohol")
         ],
-    outputs=gr.Image(type="pil"))
+    outputs=gr.outputs.Label(label="Quality")
+    )
 
 demo.launch(debug=True)