Hugging Face's logo

Spaces:
KvrParaskevi
/
Hotel-Booking-Assistant-Chat-Simple
Paused

App Files Community
KvrParaskevi commited on
Commit
bc7c6b4
·
verified ·
1 Parent(s): 3c69ea4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -1,8 +1,7 @@
1
  import gradio as gr
2
  from langchain_core.pydantic_v1 import BaseModel, Field
3
  from langchain.prompts import HumanMessagePromptTemplate, ChatPromptTemplate
4
- from langchain.output_parsers import TextIteratorStreamer
5
- from transformers import AutoModelForCausalLM, AutoTokenizer
6
 
7
  # Load the Hugging Face model and tokenizer
8
  model_name = "KvrParaskevi/Llama-2-7b-Hotel-Booking-Model"
 
1
  import gradio as gr
2
  from langchain_core.pydantic_v1 import BaseModel, Field
3
  from langchain.prompts import HumanMessagePromptTemplate, ChatPromptTemplate
4
+ from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
 
5
 
6
  # Load the Hugging Face model and tokenizer
7
  model_name = "KvrParaskevi/Llama-2-7b-Hotel-Booking-Model"