KMayanja commited on
Commit
b17ec4d
·
verified ·
1 Parent(s): 4e19e83

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -45,14 +45,14 @@ print("Model ready on:", "GPU" if torch.cuda.is_available() else "CPU")
45
 
46
  # === Translation function ===
47
  @spaces.GPU(duration=180)
48
- def predict(text, source_lang="eng", target_lang="lug"):
49
  if not text.strip():
50
  return "Please enter text to translate."
51
 
52
  langs = {"eng":"eng_Latn","lug":"lug_Latn"}
53
 
54
- src = langs.get(source_lang)
55
- tgt = langs.get(target_lang)
56
 
57
 
58
  tokenizer.src_lang = src # only needed for some NLLB versions
 
45
 
46
  # === Translation function ===
47
  @spaces.GPU(duration=180)
48
+ def predict(text, source_language="eng", target_language="lug"):
49
  if not text.strip():
50
  return "Please enter text to translate."
51
 
52
  langs = {"eng":"eng_Latn","lug":"lug_Latn"}
53
 
54
+ src = langs.get(source_language)
55
+ tgt = langs.get(target_language)
56
 
57
 
58
  tokenizer.src_lang = src # only needed for some NLLB versions