TiberiuCristianLeon commited on
Commit
7bc13dc
·
verified ·
1 Parent(s): 85f93c3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -1,6 +1,7 @@
1
  import streamlit as st
2
  from transformers import T5Tokenizer, T5ForConditionalGeneration, AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
3
  import torch
 
4
 
5
  # Create the app layout
6
  st.header("Text Machine Translation")
@@ -58,6 +59,7 @@ if submit_button:
58
  url = os.environ('GTRANSURL')
59
  params = {'sl': sselected_language, 'tl': tselected_language, 'dt': 't', 'ie': 'UTF-8', 'oe': 'UTF-8', 'q': input_text}
60
  translated_text = requests.get(url, params=params).text[0][0]
 
61
  elif model_name.startswith('t5'):
62
  prompt = f'translate {sselected_language} to {tselected_language}: {input_text}'
63
  print(prompt)
 
1
  import streamlit as st
2
  from transformers import T5Tokenizer, T5ForConditionalGeneration, AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
3
  import torch
4
+ import os
5
 
6
  # Create the app layout
7
  st.header("Text Machine Translation")
 
59
  url = os.environ('GTRANSURL')
60
  params = {'sl': sselected_language, 'tl': tselected_language, 'dt': 't', 'ie': 'UTF-8', 'oe': 'UTF-8', 'q': input_text}
61
  translated_text = requests.get(url, params=params).text[0][0]
62
+ print(translated_text.url)
63
  elif model_name.startswith('t5'):
64
  prompt = f'translate {sselected_language} to {tselected_language}: {input_text}'
65
  print(prompt)