Copain22 commited on
Commit
bede5dc
Β·
verified Β·
1 Parent(s): 4eec428

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -4,7 +4,11 @@ from pathlib import Path
4
  import torch
5
  import gradio as gr
6
  from huggingface_hub import login
7
- from langchain_community.embeddings import HuggingFaceEmbeddings # Correct import
 
 
 
 
8
  from llama_index.core import (
9
  VectorStoreIndex,
10
  SimpleDirectoryReader,
@@ -46,7 +50,6 @@ docs = SimpleDirectoryReader(
46
  input_files=[str(p) for p in Path(".").glob("*.pdf")]
47
  ).load_data()
48
 
49
- # Using the correct import from langchain_community
50
  embed_model = LangchainEmbedding(
51
  HuggingFaceEmbeddings(model_name="sentence-transformers/all-mpnet-base-v2")
52
  )
@@ -88,7 +91,7 @@ def get_chat_engine():
88
  # ──────────────────── 3. Gradio UI ────────────────────
89
  with gr.Blocks(title="CafΓ© Eleven Chat") as demo:
90
  gr.Markdown("## β˜• CafΓ© Eleven Ordering Assistant")
91
- chatbot = gr.Chatbot(height=500)
92
  user_in = gr.Textbox(placeholder="Hi, I'd like a latte...", label="Your order")
93
  clear = gr.Button("Clear Chat")
94
 
 
4
  import torch
5
  import gradio as gr
6
  from huggingface_hub import login
7
+ try:
8
+ from langchain_huggingface import HuggingFaceEmbeddings # New recommended import
9
+ except ImportError:
10
+ # Fallback to community version if new package not available
11
+ from langchain_community.embeddings import HuggingFaceEmbeddings
12
  from llama_index.core import (
13
  VectorStoreIndex,
14
  SimpleDirectoryReader,
 
50
  input_files=[str(p) for p in Path(".").glob("*.pdf")]
51
  ).load_data()
52
 
 
53
  embed_model = LangchainEmbedding(
54
  HuggingFaceEmbeddings(model_name="sentence-transformers/all-mpnet-base-v2")
55
  )
 
91
  # ──────────────────── 3. Gradio UI ────────────────────
92
  with gr.Blocks(title="CafΓ© Eleven Chat") as demo:
93
  gr.Markdown("## β˜• CafΓ© Eleven Ordering Assistant")
94
+ chatbot = gr.Chatbot(height=500, type="messages") # Fixed warning
95
  user_in = gr.Textbox(placeholder="Hi, I'd like a latte...", label="Your order")
96
  clear = gr.Button("Clear Chat")
97