Update app.py

app.py

@@ -31,7 +31,8 @@ def wine_predictor(fixed_acidity, volatile_acidity, citric_acid, residual_sugar,
    
    
     res = model.predict(df) 
-    wine_quality_url = "https://github.com/rogoran/id2223kth.github.io/blob/master/src/serverless-ml-intro/serverless-ml-lab/wine_images/wine-" + str(res[0]) + ".png"
+    wine_quality_url = "https://raw.githubusercontent.com/rogoran/id2223kth.github.io/master/src/serverless-ml-intro/serverless-ml-lab/wine_images/wine-" + str(res[0]) + ".png"
+    print(wine_quality_url)
     img = Image.open(requests.get(wine_quality_url, stream=True).raw)            
     return img
     
@@ -54,7 +55,8 @@ demo = gr.Interface(
         gr.inputs.Number(default=0.3, label="sulfur dioxide free-total ratio"),
         gr.inputs.Checkbox(label="Red wine?")
         ],
-    outputs=gr.Image(type="pil"))
+    css = ".output-image, .input-image, .image-preview {height: 600px !important}",
+    outputs=gr.Image(type="pil", width=100, height=300))
 
 demo.launch(debug=True)