Streamlit_Docker_Format_Output

Sleeping

ErikH commited on Dec 1, 2023

Commit

dab4ccd

1 Parent(s): 565faf3

Update pages/bot.py

Files changed (1) hide show

pages/bot.py CHANGED Viewed

@@ -106,7 +106,7 @@ def main():
         Question: {question}
         """
         prompt = ChatPromptTemplate.from_template(template)
-        model = "hkunlp/instructor-base"
         def format_docs(docs):

         Question: {question}
         """
         prompt = ChatPromptTemplate.from_template(template)
+        model = AutoModel.from_pretrained("hkunlp/instructor-base")
         def format_docs(docs):