Update app.py
Browse files
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
|
| 30 |
-
# We'll use the "text2text-generation" pipeline for T5-like models
|
| 31 |
-
# Model: google/flan-t5-base
|
| 32 |
generation_pipeline = pipeline(
|
| 33 |
-
"
|
| 34 |
-
model="
|
| 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):
|