|
import os |
|
|
|
import requests |
|
import streamlit as st |
|
|
|
|
|
API_URL = "https://api-inference.huggingface.co/models/andrek/nb2nn" |
|
|
|
def translate(text): |
|
headers = {"Authorization": f"Bearer {os.environ['BEARER']}"} |
|
payload = { |
|
"inputs": text, |
|
} |
|
try: |
|
response = requests.post(API_URL, headers=headers, json=payload) |
|
json_response = response.json()[0] |
|
if "error" in json_response: |
|
return f"Model still loading: {json_response}" |
|
elif "translation_text" in json_response: |
|
return json_response["translation_text"] |
|
except Exception as exc: |
|
return f"Oops, something went wrong: {exc}" |
|
|
|
|
|
st.set_page_config( |
|
page_title='Norwegian Bokmaal to Nynorsk', |
|
page_icon='translator-icon.png', |
|
) |
|
st.title("Language Translator from Norwegian Bokmaal to Nynorsk :balloon:") |
|
text = st.text_area("Enter text:", |
|
height=None, |
|
max_chars=None, |
|
key=None, |
|
help="Enter your text here", |
|
) |
|
|
|
if st.button('Translate'): |
|
if str(text).strip() == "": |
|
st.warning('Please **enter text** for translation') |
|
else: |
|
st.info(str(translate(text))) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|