Dhahlan2000 commited on
Commit
5cbe369
·
verified ·
1 Parent(s): 78591f3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -27,7 +27,7 @@ pipe = pipeline("translation", model="thilina/mt5-sinhalese-english")
27
  sin_trans_model = AutoModelForSeq2SeqLM.from_pretrained("thilina/mt5-sinhalese-english")
28
  si_trans_tokenizer = AutoTokenizer.from_pretrained("thilina/mt5-sinhalese-english")
29
 
30
- singlish_pipe = pipeline("text2text-generation", model="Dhahlan2000/Simple_Translation-model-for-GPT-v15")
31
 
32
  # Translation functions
33
  def translate_Singlish_to_sinhala(text):
@@ -116,6 +116,7 @@ def conversation_predict(input_text):
116
 
117
  def ai_predicted(user_input):
118
  user_input = translate_Singlish_to_sinhala(user_input)
 
119
  user_input = transliterate_to_sinhala(user_input)
120
  print("You(Sinhala): ", user_input,"\n")
121
  user_input = translate_sinhala_to_english(user_input)
 
27
  sin_trans_model = AutoModelForSeq2SeqLM.from_pretrained("thilina/mt5-sinhalese-english")
28
  si_trans_tokenizer = AutoTokenizer.from_pretrained("thilina/mt5-sinhalese-english")
29
 
30
+ singlish_pipe = pipeline("text2text-generation", model="Dhahlan2000/Simple_Translation-model-for-GPT-v8")
31
 
32
  # Translation functions
33
  def translate_Singlish_to_sinhala(text):
 
116
 
117
  def ai_predicted(user_input):
118
  user_input = translate_Singlish_to_sinhala(user_input)
119
+ print("You(Singlish): ", user_input,"\n")
120
  user_input = transliterate_to_sinhala(user_input)
121
  print("You(Sinhala): ", user_input,"\n")
122
  user_input = translate_sinhala_to_english(user_input)