Gabriel382 commited on
Commit
3c8bea0
·
1 Parent(s): ace96da

adding all the app files of hoverlingo

Browse files
Files changed (3) hide show
  1. Dockerfile +8 -0
  2. app.py +22 -0
  3. requirements.txt +4 -0
Dockerfile ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ FROM python:3.9
2
+
3
+ WORKDIR /app
4
+ COPY requirements.txt .
5
+ RUN pip install --no-cache-dir -r requirements.txt
6
+ COPY . .
7
+
8
+ CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
app.py ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from fastapi import FastAPI
2
+ from pydantic import BaseModel
3
+ from transformers import pipeline
4
+
5
+ # Initialize API
6
+ app = FastAPI()
7
+
8
+ # Load translation model (French)
9
+ translator = pipeline("translation", model="facebook/m2m100_418M")
10
+
11
+ # Request model
12
+ class TextRequest(BaseModel):
13
+ text: str
14
+
15
+ @app.post("/translate")
16
+ def translate_text(request: TextRequest):
17
+ translated = translator(request.text, src_lang="en", tgt_lang="fr")[0]['translation_text']
18
+ return {"translated_text": translated}
19
+
20
+ @app.get("/")
21
+ def home():
22
+ return {"message": "Translation server is running!"}
requirements.txt ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ fastapi
2
+ uvicorn
3
+ transformers
4
+ torch