Update app.py

app.py

@@ -4,11 +4,7 @@ from transformers import (
     GPT2LMHeadModel, GPT2Tokenizer,
     pipeline
 )
-
-# App title
 st.title("Multi Chatbot")
-
-# Define models and descriptions
 models = {
     "English to French": {
         "name": "Helsinki-NLP/opus-mt-en-fr",
@@ -24,7 +20,6 @@ models = {
     }
 }
 
-# Sidebar: Model selection
 st.sidebar.header("Choose a Model")
 selected_model_key = st.sidebar.radio("Select a Model:", list(models.keys()))
 model_name = models[selected_model_key]["name"]
@@ -32,7 +27,6 @@ model_description = models[selected_model_key]["description"]
 
 st.sidebar.markdown(f"### Model Description\n{model_description}")
 
-# Cache model loading for efficiency
 @st.cache_resource
 def load_english_to_french():
     tokenizer = MarianTokenizer.from_pretrained("Helsinki-NLP/opus-mt-en-fr")
@@ -50,7 +44,7 @@ def load_story_generator():
     tokenizer.pad_token = tokenizer.eos_token  # Set pad token to EOS token
     return tokenizer, model
 
-# Load the selected model
+
 if selected_model_key == "English to French":
     st.write("Loading English to French model...")
     en_fr_tokenizer, en_fr_model = load_english_to_french()
@@ -66,7 +60,6 @@ elif selected_model_key == "Story Generator":
     story_gen_tokenizer, story_gen_model = load_story_generator()
     st.write("Story Generator model loaded successfully.")
 
-# User input
 user_input = st.text_input("Enter your query:")
 
 if user_input: