k3ybladewielder commited on
Commit
4ce5313
1 Parent(s): d0c42cc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ model = "tiiuae/falcon-7b-instruct" # Use a specific model (option 1) or
6
  # model = "tiiuae/falcon-40b-instruct" # Use another specific model (option 2)
7
  get_completion = pipeline(task=task, model=model, trust_remote_code=True) # Create a text generation pipeline
8
 
9
- device = "cuda" if torch.cuda.is_available() else "cpu"
10
 
11
  # Define a function to format the chat history and create a prompt for the chatbot
12
  def format_chat_prompt(message, chat_history):
 
6
  # model = "tiiuae/falcon-40b-instruct" # Use another specific model (option 2)
7
  get_completion = pipeline(task=task, model=model, trust_remote_code=True) # Create a text generation pipeline
8
 
9
+ # device = "cuda" if torch.cuda.is_available() else "cpu"
10
 
11
  # Define a function to format the chat history and create a prompt for the chatbot
12
  def format_chat_prompt(message, chat_history):