Upload 3 files

Dockerfile

@@ -0,0 +1,29 @@
+# Use an official Python runtime as a parent image
+FROM python:3.8-slim-buster
+
+# Set the working directory in the container to /app
+WORKDIR /app
+
+# Clone the mmtafrica repository
+RUN git clone https://github.com/edaiofficial/mmtafrica.git
+
+# Add the current directory contents into the container at /app
+ADD . /app
+
+# Install virtualenv and activate it
+RUN pip install --no-cache-dir virtualenv
+
+# Install any needed packages specified in requirements.txt
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Install mmtafrica from the cloned repository
+RUN pip install --no-cache-dir /app/mmtafrica
+
+# Install additional packages
+RUN pip install --no-cache-dir numpy pandas scikit-learn matplotlib
+
+# Make port 80 available to the world outside this container
+EXPOSE 80
+
+# Run app.py when the container launches
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "80"]

app.py

@@ -0,0 +1,33 @@
+from fastapi import FastAPI
+from typing import Optional
+import torch
+from mmtafrica.mmtafrica import load_params, translate
+from huggingface_hub import hf_hub_download
+
+app = FastAPI()
+
+language_map = {'English':'en','Swahili':'sw','Fon':'fon','Igbo':'ig',
+                'Kinyarwanda':'rw','Xhosa':'xh','Yoruba':'yo','French':'fr'}
+
+# Load parameters and model from checkpoint
+checkpoint = hf_hub_download(repo_id="chrisjay/mmtafrica", filename="mmt_translation.pt")
+device = 'gpu' if torch.cuda.is_available() else 'cpu'
+params = load_params({'checkpoint':checkpoint,'device':device})
+
+@app.post("/translate")
+async def translate(data: dict):
+    source_language = data['source_language']
+    target_language = data['target_language']
+    source_sentence = data['source_sentence']
+
+    source_language_ = language_map[source_language]
+    target_language_ = language_map[target_language]
+
+    try:
+        pred = translate(params,source_sentence,source_lang=source_language_,target_lang=target_language_)
+        if pred=='':
+            return {"translation": "Could not find translation"}
+        else:
+            return {"translation": pred}
+    except Exception as error:
+        return {"error": f"Issue with translation: \n {error}"}

requirements.txt

@@ -0,0 +1,8 @@
+torch
+huggingface_hub
+fastapi
+uvicorn
+numpy
+pandas
+scikit-learn
+matplotlib