Retriever Made

README.md

@@ -3,8 +3,8 @@ title: Shantanu Datta Wikipedia
 emoji: 💬
 colorFrom: yellow
 colorTo: purple
-sdk: gradio
-sdk_version: 5.0.1
+sdk: streamlit
+sdk_version: 1.35.0
 app_file: app.py
 pinned: false
 license: apache-2.0

app.py

@@ -1,14 +1,11 @@
 import streamlit as st
-from langchain_community.document_loaders import HuggingFaceDatasetLoader
-from langchain_text_splitters import CharacterTextSplitter
-from langchain.vectorstores import Chroma
-from langchain_aws import BedrockEmbeddings
-from langchain.chat_models import ChatBedrock
+import time
+from langchain_community.vectorstores import Chroma
+from langchain_community.embeddings import BedrockEmbeddings
+from langchain_community.chat_models import ChatBedrock
+from langchain_core.documents import Document
 from langchain.schema import HumanMessage
-import os
-
-# Optional: For Cohere reranking
-import cohere
+from datasets import load_dataset
 
 # --- Load and Prepare Dataset (run once and cache)
 @st.cache_resource

requirements.txt

@@ -1,8 +1,10 @@
 huggingface_hub==0.25.2
-streamlit
-langchain
-langchain-community
-langchain-aws
-chromadb
-cohere
-boto3
+streamlit==1.32.2
+langchain==0.2.1
+langchain-core==0.1.48
+langchain-community==0.0.38
+langchain-aws==0.1.6
+chromadb==0.4.24
+boto3==1.34.82
+datasets==2.18.0
+cohere==5.3.3  # Only if you're using cohere reranker