test tansformers

app.py

@@ -1,10 +1,17 @@
 import streamlit as st
 from langchain.memory import ConversationBufferMemory
 from langchain.memory.chat_message_histories import StreamlitChatMessageHistory
+from langchain.llms.huggingface_pipeline import HuggingFacePipeline
 
 msgs = StreamlitChatMessageHistory()
 memory = ConversationBufferMemory(memory_key="history", chat_memory=msgs)
 
+hf = HuggingFacePipeline.from_model_id(
+    model_id="gpt2",
+    task="text-generation",
+    pipeline_kwargs={"max_new_tokens": 10},
+)
+
 st.title("🪩🤖")
 
 if len(msgs.messages) == 0: