Update app.py

app.py

@@ -13,7 +13,7 @@ import lightgbm as lgb
 from PIL import Image
 from flask_cors import CORS, cross_origin
 
-
+import wikipedia as wiki
 
 app = Flask(__name__)
 
@@ -52,7 +52,9 @@ def upload_image():
             index = predictions.index(max_num)
             print(classes[index])
             os.remove(str(imag.filename))
-            response = jsonify(output=classes[index])
+            result = wiki.summary(classes[index])
+            response = jsonify(output=classes[index],desc = result)
+
             # response.headers.add('Access-Control-Allow-Origin', '*')
             # response.headers.add('Access-Control-Allow-Headers', 'Content-Type,Authorization')
             # response.headers.add('Access-Control-Allow-Methods', 'GET,PUT,POST,DELETE,OPTIONS')