Update app.py

app.py

@@ -3,17 +3,18 @@ import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
 # Load model and tokenizer
-#model_name = "codellama/CodeLlama-7b-Python-hf"
+# model_name = "codellama/CodeLlama-7b-Python-hf"
 model_name = "Qwen/Qwen2.5-Coder-0.5B-Instruct"
 st.title("Python Code Helper")
 
 try:
-    st.info("Loading model... This may take a few moments.")
+    info = st.empty()
+    info.markdown("#### :red[Model is Loading....]")
     model = AutoModelForCausalLM.from_pretrained(model_name)
     tokenizer = AutoTokenizer.from_pretrained(model_name)
     device = torch.device("cuda" if torch.cuda.is_available() else 'cpu')
     model = model.to(device)
-    st.success("Model loaded successfully.")
+    info.markdown("#### :green[Model Loaded Successfully]")
 except Exception as e:
     st.error(f"Error loading model: {e}")
     st.stop()
@@ -23,7 +24,7 @@ st.markdown("### Python Code Generation")
 with st.form(key="code_form"):
     prompt = st.text_area("Enter your coding prompt:", height=200)
     submit = st.form_submit_button("Generate Code")
-    
+
     if submit and prompt.strip():
         with st.spinner("Generating response..."):
             try:
@@ -33,4 +34,4 @@ with st.form(key="code_form"):
                 st.markdown("### Generated Code:")
                 st.code(response, language="python")
             except Exception as e:
-                st.error(f"An error occurred: {e}")
+                st.error(f"An error occurred: {e}")