WSLINMSAI commited on
Commit
2925a06
·
verified ·
1 Parent(s): 173de9f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,8 +5,8 @@ from transformers import pipeline
5
  # Replace "EleutherAI/gpt-neo-2.7B" with other models like "mosaicml/mpt-7b-chat" or "OpenAssistant/oasst-sft-4-pythia-12b-epoch-3.5"
6
  generation_pipeline = pipeline(
7
  "text-generation",
8
- model="EleutherAI/gpt-neo-2.7B", # Replace this with the desired advanced model
9
- device=0 # Use GPU if available
10
  )
11
 
12
  def dental_chatbot_response(message, history):
 
5
  # Replace "EleutherAI/gpt-neo-2.7B" with other models like "mosaicml/mpt-7b-chat" or "OpenAssistant/oasst-sft-4-pythia-12b-epoch-3.5"
6
  generation_pipeline = pipeline(
7
  "text-generation",
8
+ model="EleutherAI/gpt-neo-2.7B",
9
+ device=-1 # Use CPU explicitly
10
  )
11
 
12
  def dental_chatbot_response(message, history):