File size: 549 Bytes
8ca422c
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
import streamlit as st
from rag_utils_unsloth import load_faiss_index, get_embedding_model, query_index, generate_answer

st.title("🎓 EduPilot (Unsloth + Mistral 4bit CPU)")

index, documents = load_faiss_index()
model_embed = get_embedding_model()

user_input = st.text_input("Pose ta question ici :")

if user_input:
    top_docs = query_index(user_input, index, documents, model_embed)
    context = "\n".join(top_docs)
    response = generate_answer(user_input, context)
    st.markdown("### ✨ Réponse du chatbot :")
    st.write(response)