Final: Streamlit app now uses hosted HF model

app/app.py

@@ -1,8 +1,8 @@
+
 import streamlit as st
 from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
 import torch
 
-# Load model and tokenizer
 @st.cache_resource
 def load_model():
     model = AutoModelForSeq2SeqLM.from_pretrained("Tuathe/codementor-flan")
@@ -13,19 +13,11 @@ model, tokenizer = load_model()
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model.to(device)
 
-# Streamlit page config
 st.set_page_config(page_title="CodeMentor AI", page_icon="💻", layout="centered")
 
-st.markdown(
-    "<h1 style='text-align: center;'>CodeMentor AI</h1>", 
-    unsafe_allow_html=True
-)
-st.markdown(
-    "<p style='text-align: center; font-size:18px;'>Your AI Coding Interview Assistant</p>",
-    unsafe_allow_html=True
-)
+st.markdown("<h1 style='text-align: center;'>CodeMentor AI</h1>", unsafe_allow_html=True)
+st.markdown("<p style='text-align: center; font-size:18px;'>Your AI Coding Interview Assistant</p>", unsafe_allow_html=True)
 
-# Sidebar info
 with st.sidebar:
     st.title("About CodeMentor AI")
     st.info(
@@ -35,7 +27,6 @@ with st.sidebar:
     st.markdown("---")
     st.markdown("Created by Chetan")
 
-# Chat interface
 user_input = st.text_area("Ask your coding question here:", height=150)
 
 if st.button("Get Answer"):