KvrParaskevi commited on
Commit
ee49276
·
verified ·
1 Parent(s): 2079e61

Update chatbot.py

Browse files
Files changed (1) hide show
  1. chatbot.py +1 -0
chatbot.py CHANGED
@@ -2,6 +2,7 @@ import os
2
  from langchain.memory import ConversationBufferMemory
3
  from langchain.chains import ConversationChain
4
  import langchain.globals
 
5
  from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig, pipeline
6
  import streamlit as st
7
 
 
2
  from langchain.memory import ConversationBufferMemory
3
  from langchain.chains import ConversationChain
4
  import langchain.globals
5
+ from langchain_community.llms.huggingface_pipeline import HuggingFacePipeline
6
  from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig, pipeline
7
  import streamlit as st
8