Create app.py

app.py

@@ -0,0 +1,28 @@
+from transformers import MarianMTModel, MarianTokenizer
+from flask import Flask, request, jsonify
+
+app = Flask(__name__)
+
+# Load pre-trained model and tokenizer
+model_name = 'Helsinki-NLP/opus-mt-en-fr'
+model = MarianMTModel.from_pretrained(model_name)
+tokenizer = MarianTokenizer.from_pretrained(model_name)
+
+@app.route('/translate', methods=['POST'])
+def translate():
+    input_text = request.json['text']
+    
+    # Tokenize input text
+    inputs = tokenizer(input_text, return_tensors='pt', padding=True, truncation=True)
+
+    # Perform translation
+    with torch.no_grad():
+        translated = model.generate(**inputs)
+
+    # Decode translated text
+    translated_text = tokenizer.batch_decode(translated, skip_special_tokens=True)[0]
+
+    return jsonify({'translation': translated_text})
+
+if __name__ == '__main__':
+    app.run(debug=True)