fix

app.py

@@ -1,7 +1,26 @@
-from fastapi import FastAPI
+from flask import Flask, request, jsonify
+from transformers import pipeline
 
-app = FastAPI()
+app = Flask(__name__)
 
-@app.get("/")
-def greet_json():
-    return {"Hello": "World!"}
+translator = pipeline("translation", model="Bildad/Swahili-English_Translation")
+
+
+@app.route("/translate", methods=["POST"])
+def translate():
+    data = request.get_json()
+
+    if "text" not in data:
+        return jsonify({"error": "No text provided"}), 400
+
+    text_to_translate = data["text"]
+    translation = translator(text_to_translate)[0]
+    translated_text = translation["translation_text"]
+
+    return jsonify({"translated_text": translated_text})
+
+
+if __name__ == "__main__":
+    from waitress import serve
+
+    serve(app, host="0.0.0.0", port=10000)

requirements.txt

@@ -1,13 +1,35 @@
-annotated-types==0.7.0
-anyio==4.4.0
+blinker==1.8.2
+certifi==2024.7.4
+charset-normalizer==3.3.2
 click==8.1.7
-exceptiongroup==1.2.2
-fastapi==0.112.2
-h11==0.14.0
+filelock==3.15.4
+Flask==3.0.3
+fsspec==2024.6.1
+huggingface-hub==0.24.6
 idna==3.8
-pydantic==2.8.2
-pydantic_core==2.20.1
-sniffio==1.3.1
-starlette==0.38.2
+itsdangerous==2.2.0
+Jinja2==3.1.3
+joblib==1.4.2
+MarkupSafe==2.1.5
+mpmath==1.3.0
+networkx==3.2.1
+numpy==2.1.0
+packaging==24.1
+pillow==10.2.0
+PyYAML==6.0.2
+regex==2024.7.24
+requests==2.32.3
+sacremoses==0.1.1
+safetensors==0.4.4
+sentencepiece==0.2.0
+sympy==1.12
+tokenizers==0.19.1
+torch==2.4.0
+torchaudio==2.4.0
+torchvision==0.19.0
+tqdm==4.66.5
+transformers==4.44.2
 typing_extensions==4.12.2
-uvicorn==0.30.6
+urllib3==2.2.2
+waitress==3.0.0
+Werkzeug==3.0.4%