Update app.py

app.py

@@ -1,5 +1,6 @@
 import streamlit as st
 from transformers import MarianMTModel, MarianTokenizer
+import torch
 
 # Title and Description
 st.set_page_config(page_title='Language Translation App', layout='centered')
@@ -9,6 +10,8 @@ st.write('Translate text between multiple languages using an open-source model.'
 # Model Selection
 @st.cache_resource
 def load_model(src_lang, tgt_lang):
+    if not torch.cuda.is_available():
+        st.warning('CUDA is not available. The model will run on CPU, which may be slower.')
     model_name = f'Helsinki-NLP/opus-mt-{src_lang}-{tgt_lang}'
     tokenizer = MarianTokenizer.from_pretrained(model_name)
     model = MarianMTModel.from_pretrained(model_name)
@@ -16,8 +19,8 @@ def load_model(src_lang, tgt_lang):
 
 # Sidebar for Language Selection
 st.sidebar.header('Language Selection')
-src_lang = st.sidebar.selectbox('Select Source Language', ['en', 'fr', 'es', 'de', 'it','pt','ru','zh','ja','ar'], index=0)
-tgt_lang = st.sidebar.selectbox('Select Target Language', ['en', 'fr', 'es', 'de', 'it','pt','ru','zh','ja','ar'], index=1)
+src_lang = st.sidebar.selectbox('Select Source Language', ['en', 'fr', 'es', 'de', 'it', 'pt', 'ru', 'zh', 'ja', 'ar'], index=0)
+tgt_lang = st.sidebar.selectbox('Select Target Language', ['en', 'fr', 'es', 'de', 'it', 'pt', 'ru', 'zh', 'ja', 'ar'], index=1)
 
 # Main Input Area
 st.write('### Enter text to translate:')
@@ -42,4 +45,6 @@ if st.button('Translate'):
 st.sidebar.write('Powered by Helsinki-NLP/opus-mt')
 st.sidebar.write('Built with Streamlit')
 
-# To run the app, save this file and run `streamlit run filename.py` in your terminal.
+# Instructions for Running
+st.write('**To run the app:**')
+st.code('streamlit run app.py')