Update app.py

app.py

@@ -30,20 +30,7 @@ def embed(text):
         outputs = steelbert_model(**inputs, output_hidden_states=True)
     return outputs.hidden_states[-1][:,0,:].cpu().numpy()[0]
 
-# --- 3. Base documentaire (exemple, à remplacer par tes documents techniques) ---
-docs = {
-    "doc1": "L’acier X42 a une résistance à la traction de 415 MPa.",
-    "doc2": "L’acier inoxydable 304 est résistant à la corrosion et à l’oxydation."
-}
-doc_embeddings = {k: embed(v) for k,v in docs.items()}
-
-def search_best_doc(question):
-    q_emb = embed(question)
-    def cosine(a,b): return np.dot(a,b)/(np.linalg.norm(a)*np.linalg.norm(b))
-    best_doc = max(docs, key=lambda k: cosine(q_emb, doc_embeddings[k]))
-    return docs[best_doc]
-
-# --- 4. Fonction de réponse avec Mistral 7B Instruct ---
+# --- 3. Fonction de réponse avec Mistral 7B Instruct ---
 def respond(
     message,
     history: list[dict[str, str]],
@@ -80,7 +67,7 @@ def respond(
             response += token
             yield response
 
-# --- 5. Interface Gradio ---
+# --- 4. Interface Gradio ---
 chatbot = gr.ChatInterface(
     respond,
     type="messages",