Update src/streamlit_app.py

src/streamlit_app.py

@@ -4,7 +4,7 @@ import numpy as np
 import pandas as pd
 from sentence_transformers import SentenceTransformer
 import pickle
-from langchain_ollama import ChatOllama
+from langchain_huggingface import HuggingFaceEndpoint
 from transformers import AutoTokenizer, AutoModel
 import torch
 
@@ -63,8 +63,11 @@ def get_ollama_explanation(query, description):
     response = llm.invoke(prompt)
     return response.content.strip()
 
-# Load Ollama
-llm = ChatOllama(model="llama3.2")
+# Load LLM
+llm = HuggingFaceEndpoint(
+    repo_id="mistralai/Mistral-7B-Instruct-v0.3",
+    huggingfacehub_api_token=os.environ["HF_TOKEN"]
+)
 
 # Initialize app
 st.set_page_config(page_title="Fragrance Recommendation System", layout="wide")