Initial commit

README.md

@@ -0,0 +1,33 @@
+# Comparison of Machine-Translations Models
+
+<div align="center">
+ <p>
+    <img style="" src="./logounipi.png" alt="Logo" width="250" >  <br>
+  </p>
+</div>
+<br>
+<div align="center">
+ <p align="center"><h3>Authors</h3>
+    <a href="mailto:[email protected]">Gennaro Daniele Acciaro</a>
+    ·
+    <a href="mailto:@studenti.unipi.it"> - - </a>   
+    ·
+    <a href="mailto:@studenti.unipi.it"> - - </a>
+  </p>
+    <p align="center">
+    <h3><a href="./report.pdf">Report</a></h3>
+  </p>
+    <!-- <p align="center">
+            <h3><a href="./slides.pdf">Slides</a></h3>
+          </p>
+        -->
+</div>
+
+Master Degree (Artificial Intelligence curriculum)<br>
+**HLT** course, Academic Year: 2021/2022<br>
+Date: May 2022<br>
+
+## Description
+- x 
+### Abstract
+- c

app.py

@@ -0,0 +1,53 @@
+from flask import Flask, request, jsonify
+from flask_cors import CORS, cross_origin
+import time
+
+print("[SERVER] Server loading...")
+import_start_time = time.time()
+
+from controller.FS_TransformersController import fs_transformers_translate
+from controller.FS_LSTMController import fs_lstm_translate
+from controller.DeepL_Controller import deepl_translate
+from controller.Helsinki_Controller import helsinki_translate
+from controller.IBMModel_Controller import ibm_translate
+from controller.T5_Controller import t5_translate
+
+app = Flask(__name__, static_url_path='/static')
+app.config['CORS_HEADERS'] = 'Content-Type'
+cors = CORS(app)
+
+print("[SERVER] Server loaded in ", time.time()-import_start_time, " seconds")
+
+@app.route('/')
+def hello_world():  # put application's code here
+    return "render_template('index.html')"
+
+
+@cross_origin()
+@app.route('/translate')
+def query_example():
+    model = request.args.get('model')
+    sentence = request.args.get('sentence')
+
+    if model == 'IBM Model 1 (50k)' or model == 'IBM Model 1':
+        return jsonify({"response": ibm_translate(sentence)})
+    if model == 'LSTM (Custom)':
+        return jsonify({"response": fs_lstm_translate(sentence)})
+    if model == 'Transformer (Custom)':
+        return jsonify({"response": fs_transformers_translate(sentence)})
+    if model == 'T5':
+        return jsonify({"response": t5_translate(sentence)})
+    #if model == 'Bert2Bert':
+    #    return jsonify({"response": fs_lstm_translate(sentence)})
+    #if model == 'T52Bert':
+    #    return jsonify({"response": fs_lstm_translate(sentence)})
+    if model == 'DeepL':
+        return jsonify({"response": deepl_translate(sentence)})
+    if model == 'Helsinki':
+        return jsonify({"response": helsinki_translate(sentence)})
+
+    return jsonify({"response": "error"})
+
+
+if __name__ == '__main__':
+    app.run()