Upload 4 files

Dockerfile

@@ -0,0 +1,12 @@
+FROM python:3.10-slim
+
+WORKDIR /app
+
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+COPY app.py .
+
+EXPOSE 7860
+
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

README.md

@@ -1,12 +1,34 @@
 ---
-title: Ga En
-emoji: 👀
-colorFrom: gray
-colorTo: blue
+title: Irish English Translation API
+emoji: 🌍
+colorFrom: green
+colorTo: yellow
 sdk: docker
 pinned: false
-license: mit
-short_description: translation api
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# Irish-English Translation API
+
+FastAPI backend for Irish (Gaeilge) to English translation.
+
+## Endpoints
+
+- `GET /` - Health check
+- `POST /translate` - Translate text
+
+## Request format
+
+```json
+{
+  "text": "Dia dhuit",
+  "direction": "ga-en"
+}
+```
+
+## Response format
+
+```json
+{
+  "translation": "Hello"
+}
+```

app.py

@@ -0,0 +1,45 @@
+from fastapi import FastAPI
+from fastapi.middleware.cors import CORSMiddleware
+from pydantic import BaseModel
+from transformers import pipeline
+
+app = FastAPI()
+
+# Allow CORS for your website
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],  # Update with your actual domain in production
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+# Load translation models
+# Using Helsinki-NLP models - you can replace with your own fine-tuned models
+ga_en_translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ga-en")
+en_ga_translator = pipeline("translation", model="Helsinki-NLP/opus-mt-en-ga")
+
+
+class TranslationRequest(BaseModel):
+    text: str
+    direction: str  # "ga-en" or "en-ga"
+
+
+class TranslationResponse(BaseModel):
+    translation: str
+
+
+@app.get("/")
+def read_root():
+    return {"status": "Irish-English Translation API is running"}
+
+
+@app.post("/translate", response_model=TranslationResponse)
+def translate(request: TranslationRequest):
+    if request.direction == "ga-en":
+        result = ga_en_translator(request.text, max_length=512)
+    else:
+        result = en_ga_translator(request.text, max_length=512)
+
+    translation = result[0]["translation_text"]
+    return TranslationResponse(translation=translation)

requirements.txt

@@ -0,0 +1,6 @@
+fastapi
+uvicorn[standard]
+transformers
+torch
+sentencepiece
+sacremoses