moslem commited on
Commit
ef9f146
·
verified ·
1 Parent(s): 636341a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -8
app.py CHANGED
@@ -1,21 +1,15 @@
1
  from fastapi import FastAPI, Query
2
  from transformers import pipeline
3
 
4
- # ساخت اپ FastAPI
5
  app = FastAPI(title="English–Persian Translator")
6
 
7
- # لود مدل سبک ترجمه (عمومی و بدون نیاز به لاگین)
8
- translator_en_fa = pipeline(
9
- "translation_en_to_fa",
10
- model="Helsinki-NLP/opus-mt-en-fa",
11
- trust_remote_code=True
12
- )
13
 
14
  @app.get("/")
15
  def home():
16
  return {"message": "Welcome to English–Persian Translator!"}
17
 
18
  @app.get("/translate")
19
- def translate(text: str = Query(..., description="English text to translate to Persian")):
20
  result = translator_en_fa(text)
21
  return {"input": text, "translation": result[0]["translation_text"]}
 
1
  from fastapi import FastAPI, Query
2
  from transformers import pipeline
3
 
 
4
  app = FastAPI(title="English–Persian Translator")
5
 
6
+ translator_en_fa = pipeline("translation_en_to_fa", model="Helsinki-NLP/opus-mt-en-fa")
 
 
 
 
 
7
 
8
  @app.get("/")
9
  def home():
10
  return {"message": "Welcome to English–Persian Translator!"}
11
 
12
  @app.get("/translate")
13
+ def translate(text: str = Query(...)):
14
  result = translator_en_fa(text)
15
  return {"input": text, "translation": result[0]["translation_text"]}