Commit
•
2079e61
1
Parent(s):
5349598
Update chatbot.py
Browse files- chatbot.py +1 -1
chatbot.py
CHANGED
|
@@ -2,7 +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
|
| 6 |
import streamlit as st
|
| 7 |
|
| 8 |
my_model_id = os.getenv('MODEL_REPO_ID', 'Default Value')
|
|
|
|
| 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 |
|
| 8 |
my_model_id = os.getenv('MODEL_REPO_ID', 'Default Value')
|