Update app.py

app.py

@@ -7,6 +7,7 @@ import pandas as pd
 import numpy as np
 import pydotplus
 import gradio as gr
+from PIL import Image
 
 
 def Tree_Detection(sample):
@@ -51,7 +52,9 @@ def Tree_Detection(sample):
 
     result = f'The fault type is:{clf.predict([sample])[0]}'										# 预测
 
-    return result
+    image = Image.open("tree.jpg")
+
+    return result， image
 # print(Tree_Detection([2, 1, 1, 0]))
 
 def test(image):
@@ -61,7 +64,7 @@ def test(image):
 demo = gr.Interface(
 	fn=Tree_Detection,
 	inputs='text',
-	outputs='text'
+	outputs=['text', 'image']
 )
 demo.launch()