Update app.py

app.py

@@ -11,7 +11,7 @@ from operator import itemgetter
 from huggingface_hub import HfApi, HfFolder
 import gradio as gr
 from huggingface_hub import login, InferenceClient
-
+from langchain.embeddings import HuggingFaceEmbeddings
 
 
 USE_HF = True
@@ -39,8 +39,8 @@ if USE_HF:
         offload_folder="offload",  # Specify the offload folder
         token=huggingface_token
     )
-    embeddings = OllamaEmbeddings(model=MODEL_NAME)
-    lang_model = OllamaLLM(model=MODEL_NAME)
+    embeddings = HuggingFaceEmbeddings(model_name=MODEL_NAME)
+    lang_model = model
 else:
     print("Using local Ollama model...")
     MODEL = "jsk/bio-mistral"