Update app.py
Browse files
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)
|