hadheedo commited on
Commit
916f387
·
verified ·
1 Parent(s): 5cc0653

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -17,6 +17,7 @@ def translate_text(data: Prompt):
17
  global translator
18
  # تحميل الموديل أول مرة يجي request
19
  if translator is None:
20
- translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ar-en", device=-1)
21
- result = translator(data.user_prompt)
22
  return {"translation": result[0]['translation_text']}
 
 
17
  global translator
18
  # تحميل الموديل أول مرة يجي request
19
  if translator is None:
20
+ translator = pipeline("translation", model="facebook/m2m100_418M", device=-1)
21
+ result = translator(data.user_prompt, src_lang="ar", tgt_lang="en")
22
  return {"translation": result[0]['translation_text']}
23
+