Update app.py

app.py

@@ -28,7 +28,9 @@ def wine(fixed_acidity, citric_acid, type_white, chlorides, volatile_acidity, de
     # the first element.
     #     print("Res: {0}").format(res)
     print(res)
-    return str(res[0])
+    star_url = "https://raw.githubusercontent.com/SamuelHarner/review-images/main/images/" + str(res[0]) + "_stars.png"
+    img = Image.open(requests.get(star_url, stream=True).raw)
+    return img
 
 demo = gr.Interface(
     fn=wine,
@@ -45,7 +47,7 @@ demo = gr.Interface(
         gr.inputs.Number(default=0.99, label="density"),
         gr.inputs.Number(default=0.06, label="chlorides"),
     ],
-    outputs="text")
+    outputs=gr.Image(type="pil"))
 
 demo.launch(debug=True)