Add image output to wine quality prediction

app.py

@@ -2,6 +2,8 @@ import gradio as gr
 import hopsworks
 import joblib
 import pandas as pd
+from PIL import Image
+import requests
 
 project = hopsworks.login()
 fs = project.get_feature_store()
@@ -30,7 +32,11 @@ def wine(fixed_acidity, volatile_acidity, citric_acid, residual_sugar, chlorides
     # the first element.
 
     print(res)
-    return res[0]
+
+    url = "https://raw.githubusercontent.com/pierrelefevre/scalable-ml/main/lab1/task2/img/" + int(res[0]) + ".png"
+    img = Image.open(requests.get(url, stream=True).raw)            
+
+    return [res[0], img]
 
 iface = gr.Interface(
     fn=wine, 
@@ -49,6 +55,7 @@ iface = gr.Interface(
         gr.Number(label="alcohol", value=13.1),
         gr.Radio(["red", "white"], label="type", value="white")
         ],
-    outputs=gr.Number(label="quality"))
+    outputs=[gr.Number(label="quality"), 
+             gr.Image(type="pil")])
 
 iface.launch()