SucheRAG

Sleeping

alexkueck commited on Jul 1, 2024

Commit

ae0bb48

verified ·

1 Parent(s): e7e4a86

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,7 +1,7 @@
 import os
 import gradio as gr
 from langchain_community.vectorstores import Chroma
-from transformers import RagTokenizer, RagSequenceForGeneration
 from sentence_transformers import SentenceTransformer
 from langchain import LLMChain, PromptTemplate
 from langchain_community.llms import HuggingFacePipeline

 import os
 import gradio as gr
 from langchain_community.vectorstores import Chroma
+from transformers import RagTokenizer, RagSequenceForGeneration, RagRetriever
 from sentence_transformers import SentenceTransformer
 from langchain import LLMChain, PromptTemplate
 from langchain_community.llms import HuggingFacePipeline