From 8fd9af66e9eba60e3dbf5cb07a6ce599c92b7565 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Tue, 9 Apr 2024 00:58:24 +0200 Subject: Fix bug --- rag/ui.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'rag') diff --git a/rag/ui.py b/rag/ui.py index 5083bb4..37ab664 100644 --- a/rag/ui.py +++ b/rag/ui.py @@ -3,7 +3,7 @@ import streamlit as st from langchain_community.document_loaders.blob_loaders import Blob from dotenv import load_dotenv -from generator import get_generator, MODELS +from rag.generator import get_generator, MODELS from generator.prompt import Prompt from retriever.retriever import Retriever @@ -65,4 +65,4 @@ if __name__ == "__main__": with result_column: generator = get_generator(model) st.markdown("### Answer") - st.write_stream(generator.generate(generator, prompt)) + st.write_stream(generator.generate(prompt)) -- cgit v1.2.3-70-g09d2