Create app.py

app.py

@@ -18,7 +18,10 @@ if task == "Translation":
         ["Helsinki-NLP/opus-mt-en-fr", "Helsinki-NLP/opus-mt-fr-en"]
     )
     st.sidebar.write(f"Selected model: {model_name}")
-    translator = pipeline("translation", model=model_name)
+    try:
+        translator = pipeline("translation", model=model_name)
+    except Exception as e:
+        st.error(f"Error loading translation model: {e}")
 
 elif task == "Text Generation":
     model_name = st.sidebar.selectbox(
@@ -26,7 +29,10 @@ elif task == "Text Generation":
         ["gpt2", "EleutherAI/gpt-neo-125M"]
     )
     st.sidebar.write(f"Selected model: {model_name}")
-    generator = pipeline("text-generation", model=model_name)
+    try:
+        generator = pipeline("text-generation", model=model_name)
+    except Exception as e:
+        st.error(f"Error loading text generation model: {e}")
 
 elif task == "Sentiment Analysis":
     model_name = st.sidebar.selectbox(
@@ -34,7 +40,10 @@ elif task == "Sentiment Analysis":
         ["distilbert-base-uncased-finetuned-sst-2-english"]
     )
     st.sidebar.write(f"Selected model: {model_name}")
-    analyzer = pipeline("sentiment-analysis", model=model_name)
+    try:
+        analyzer = pipeline("sentiment-analysis", model=model_name)
+    except Exception as e:
+        st.error(f"Error loading sentiment analysis model: {e}")
 
 # Main app functionality
 st.header(f"{task} Task")
@@ -42,21 +51,30 @@ st.header(f"{task} Task")
 if task == "Translation":
     text = st.text_area("Enter text to translate", "Hello, how are you?")
     if st.button("Translate"):
-        translation = translator(text)
-        st.write("Translation:")
-        st.write(translation[0]['translation_text'])
+        try:
+            translation = translator(text)
+            st.write("Translation:")
+            st.write(translation[0]['translation_text'])
+        except Exception as e:
+            st.error(f"Error during translation: {e}")
 
 elif task == "Text Generation":
     prompt = st.text_area("Enter a prompt", "Once upon a time")
     max_length = st.slider("Max length", 10, 100, 50)
     if st.button("Generate Text"):
-        generated_text = generator(prompt, max_length=max_length)
-        st.write("Generated Text:")
-        st.write(generated_text[0]['generated_text'])
+        try:
+            generated_text = generator(prompt, max_length=max_length)
+            st.write("Generated Text:")
+            st.write(generated_text[0]['generated_text'])
+        except Exception as e:
+            st.error(f"Error during text generation: {e}")
 
 elif task == "Sentiment Analysis":
     text = st.text_area("Enter text for sentiment analysis", "I love learning new languages!")
     if st.button("Analyze Sentiment"):
-        sentiment = analyzer(text)
-        st.write("Sentiment Analysis Result:")
-        st.write(sentiment[0])
+        try:
+            sentiment = analyzer(text)
+            st.write("Sentiment Analysis Result:")
+            st.write(sentiment[0])
+        except Exception as e:
+            st.error(f"Error during sentiment analysis: {e}")