WSLINMSAI commited on
Commit
f3ce746
·
verified ·
1 Parent(s): d2ee098

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -5
app.py CHANGED
@@ -26,12 +26,10 @@ dental_terms = {
26
  "occlusion": "Occlusion refers to the alignment and contact between teeth when the jaws close."
27
  }
28
 
29
- # Set up a Flan-T5 pipeline (instruction-tuned T5)
30
- # We'll use the "text2text-generation" pipeline for T5-like models
31
- # Model: google/flan-t5-base
32
  generation_pipeline = pipeline(
33
- "text2text-generation",
34
- model="google/flan-t5-base"
35
  )
36
 
37
  def chatbot_response(message, history):
 
26
  "occlusion": "Occlusion refers to the alignment and contact between teeth when the jaws close."
27
  }
28
 
29
+ # Set up a gpt2-large pipeline
 
 
30
  generation_pipeline = pipeline(
31
+ "text-generation",
32
+ model="gpt2-large"
33
  )
34
 
35
  def chatbot_response(message, history):