Jaal047 commited on
Commit
f257a89
·
verified ·
1 Parent(s): fc670b4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -3
app.py CHANGED
@@ -1,11 +1,15 @@
1
  import os
2
  import gradio as gr
3
  from langchain_community.vectorstores import FAISS
4
- from langchain.embeddings import HuggingFaceEmbeddings
5
  from groq import Groq
6
 
7
- # Load FAISS index
8
- vector_store = FAISS.load_local("faiss_index/robohome_faiss", HuggingFaceEmbeddings()) # Pastikan path benar
 
 
 
 
9
 
10
  # Load API Key dari variabel lingkungan
11
  GROQ_API_KEY = os.getenv("GROQ_API_KEY")
 
1
  import os
2
  import gradio as gr
3
  from langchain_community.vectorstores import FAISS
4
+ from langchain_community.embeddings import HuggingFaceEmbeddings
5
  from groq import Groq
6
 
7
+ # Load FAISS index dengan deserialization yang aman
8
+ vector_store = FAISS.load_local(
9
+ "robohome_faiss",
10
+ HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2"),
11
+ allow_dangerous_deserialization=True
12
+ )
13
 
14
  # Load API Key dari variabel lingkungan
15
  GROQ_API_KEY = os.getenv("GROQ_API_KEY")