Update app.py
Browse files
app.py
CHANGED
|
@@ -53,7 +53,7 @@ def cache_answer(question: str, answer: str):
|
|
| 53 |
|
| 54 |
|
| 55 |
# --- Model Setup ---
|
| 56 |
-
MODEL_NAME = 'EleutherAI/gpt-neo-2.7B'#'distilbert/distilgpt2'#'deepseek-ai/DeepSeek-R1-Distill-Qwen-7B'#'mistralai/Mistral-7B-Instruct-v0.2'
|
| 57 |
|
| 58 |
def load_model():
|
| 59 |
"""Download and load the model and tokenizer."""
|
|
|
|
| 53 |
|
| 54 |
|
| 55 |
# --- Model Setup ---
|
| 56 |
+
MODEL_NAME = 'microsoft/DialoGPT-small'# 'EleutherAI/gpt-neo-2.7B'#'distilbert/distilgpt2'#'deepseek-ai/DeepSeek-R1-Distill-Qwen-7B'#'mistralai/Mistral-7B-Instruct-v0.2'
|
| 57 |
|
| 58 |
def load_model():
|
| 59 |
"""Download and load the model and tokenizer."""
|