Update app.py
Browse files
app.py
CHANGED
@@ -3,7 +3,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
|
|
3 |
from langchain_huggingface import HuggingFacePipeline # Updated import
|
4 |
from langchain_core.prompts import PromptTemplate
|
5 |
from langchain_core.runnables import RunnableLambda
|
6 |
-
from
|
7 |
|
8 |
# Load model and tokenizer
|
9 |
model_name = "microsoft/DialoGPT-medium"
|
|
|
3 |
from langchain_huggingface import HuggingFacePipeline # Updated import
|
4 |
from langchain_core.prompts import PromptTemplate
|
5 |
from langchain_core.runnables import RunnableLambda
|
6 |
+
from langchain.memory import ConversationBufferMemory
|
7 |
|
8 |
# Load model and tokenizer
|
9 |
model_name = "microsoft/DialoGPT-medium"
|