Hugging Face's logo

Spaces:
KvrParaskevi
/
Hotel-Booking-Assistant
Paused

App Files Community
1
KvrParaskevi commited on
Commit
2079e61
·
verified ·
1 Parent(s): 5349598

Update chatbot.py

Browse files
Files changed (1) hide show
  1. 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')