Update app.py

app.py

@@ -11,7 +11,6 @@ from PIL import Image
 from huggingface_hub import hf_hub_download
 from onnxruntime import InferenceSession
 import io
-from flask import Flask, request, jsonify
 
 # noinspection PyUnresolvedReferences
 def make_square(img, target_size):
@@ -164,25 +163,6 @@ def image_to_wd14_tags(image:Image.Image, link:str, model_name: str, threshold:
 
     return ratings, output_text, filtered_tags
 
-app = Flask(__name__)
-
-@app.route('/api/tag_image', methods=['POST'])
-def tag_image():
-    data = request.get_json()
-
-    input_image = data.get('input_image', None)
-    input_text = data.get('url', None)
-    model = data.get('model', 'wd14-vit-v2')
-    threshold = data.get('threshold', 0.35)
-    space = data.get('space', True)
-    escape = data.get('escape', True)
-    confidence = data.get('confidence', False)
-    order = data.get('order', True)
-
-    result = image_to_wd14_tags(input_image, input_text, model, threshold, space, escape, confidence, order)
-
-    return jsonify(result)
-
 if __name__ == '__main__':
     with gr.Blocks() as demo:
         with gr.Row():
@@ -215,5 +195,4 @@ if __name__ == '__main__':
             outputs=[gr_ratings, gr_output_text, gr_tags],
         )
 
-    app.run(host='0.0.0.0', port=5000, threaded=True)
-    demo.queue(os.cpu_count()).launch()
+        demo.queue(os.cpu_count()).launch()