Update app.py

app.py

@@ -48,7 +48,12 @@ def predict_price(year, month, markets, food):
  
   prediction = model.predict(input_data)
 
-  return prediction
+  food_url = "https://raw.githubusercontent.com/TimiUU/wine/main" + food + ".png"
+  img = Image.open(requests.get(food_url, stream=True).raw)
+ 
+  return prediction, img  
+
+  #return prediction, food_url
 
 
 demo = gr.Interface(
@@ -61,6 +66,7 @@ demo = gr.Interface(
         gr.Dropdown(choices=market, label="Market"),
         gr.Dropdown(choices=commodity, label="Food Type")
         ],
-    outputs="text"
+    outputs=[gr.Textbox(), gr.Image()]
+    
 )
 demo.launch(debug=True)