initial

.gitignore

@@ -0,0 +1 @@
+__pycache__

Dockerfile

@@ -7,4 +7,5 @@ RUN apt-get install git-lfs && git lfs install
 RUN git clone https://huggingface.co/sentence-transformers/quora-distilbert-multilingual model
 COPY . .
 ENV PYTHONPATH "${PYTHONPATH}:/code/app/"
-CMD ["uwsgi", "--http", "0.0.0.0:7860", "--master", "-p", "1", "-w", "app.main:app"]
+# CMD ["uwsgi", "--http", "0.0.0.0:7860", "--master", "-p", "1", "-w", "app.main:app"]
+CMD ["python3", "app/main.py"]

app/main.py

@@ -1,9 +1,10 @@
 import sys
 import logging
-from flask import Flask, jsonify, request
+import gradio as gr
+# from flask import Flask, jsonify, request
 from similarity import LanguageModel, Similarity
 
-app = Flask(__name__)
+# app = Flask(__name__)
 
 PRE_TRAINED_MODEL_PATH = './model'
 
@@ -18,16 +19,31 @@ init_logger()
 lm = LanguageModel(pre_trained_model_path=PRE_TRAINED_MODEL_PATH, max_len=1000)
 similarity = Similarity(featurize_fn=lm.featurize)
 
-@app.route('/getSimilarity/', methods=['GET', 'POST'])
-def process_request():
-    text1 = request.values.get('text1')
-    text2 = request.values.get('text2')
-    score = similarity.get_score(text1, text2)
+# @app.route('/getSimilarity/', methods=['GET', 'POST'])
+# def process_request():
+#     text1 = request.values.get('text1')
+#     text2 = request.values.get('text2')
+#     score = similarity.get_score(text1, text2)
+
+#     response = {'similarity score': score}
+#     response = jsonify(response)
+#     response.headers.add("Access-Control-Allow-Origin", "*")
+#     return response
 
+def process(text1, text2):
+    score = similarity.get_score(text1, text2)
     response = {'similarity score': score}
-    response = jsonify(response)
-    response.headers.add("Access-Control-Allow-Origin", "*")
     return response
+
+demo = gr.Interface(
+    fn=process,
+    inputs=[
+        gr.Textbox(lines=2),
+        gr.Textbox(lines=2),
+    ],
+    outputs=gr.JSON(),
+)
     
 if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=7860)
+    # app.run(host="0.0.0.0", port=7860)
+    demo.launch(share=False, server_name="0.0.0.0", server_port=7860)

requirements.txt

@@ -1,7 +1,8 @@
-Werkzeug==2.3.7
-Flask==2.2.0
+# Werkzeug==2.3.7
+# Flask==2.2.0
 torch==2.0.1
 numpy==1.23.5
 transformers==4.34.0
 # uvicorn==0.23.2
-uWSGI==2.0.21
+# uWSGI==2.0.21
+gradio==3.47.1