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."""
|