Duplicate from bsenst/flask_inference_api

Co-authored-by: Benjamin Senst <[email protected]>

.gitattributes

@@ -0,0 +1,27 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bin.* filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zstandard filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,14 @@
+---
+title: Flask_test
+emoji: 🦀
+colorFrom: blue
+colorTo: pink
+sdk: gradio
+sdk_version: 2.8.13
+app_file: app.py
+pinned: false
+license: mit
+duplicated_from: bsenst/flask_inference_api
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces#reference

app.py

@@ -0,0 +1,26 @@
+import flask
+from flask import request
+import os
+from dotenv import load_dotenv
+load_dotenv()
+
+app = flask.Flask(__name__, template_folder="./")
+
+from transformers import pipeline
+
+classifier = pipeline('text-classification', model="bsenst/classify_services_model")
+
+@app.route('/')
+def index():
+    return flask.render_template('index.html')
+
+@app.route("/", methods=["POST"])
+def predict():
+    incoming = request.get_json()
+    print(incoming)
+    prediction = classifier(incoming["text"])[0]
+    print(prediction)
+    return prediction
+
+if __name__ == '__main__':
+    app.run(host='0.0.0.0',  port=int(os.environ.get('PORT', 7860)))

index.html

@@ -0,0 +1,14 @@
+<h1>
+  Flask Space
+</h1>
+
+<p>
+  Welcome! This is a Flask Space! To create one, follow the next steps:
+</p>
+
+<ol>
+  <li>Create a <a href="https://huggingface.co/new-space">new Space</a></li>
+  <li>Select Gradio SDK</li>
+  <li>Add Flask to the requirements</li>
+  <li>Create your Flask app in app.py file</li>
+</ol>

requirements.txt

@@ -0,0 +1,4 @@
+Flask==2.0.2
+python-dotenv==0.19.2
+transformers
+torch