magdap116 commited on
Commit
f91ee98
·
verified ·
1 Parent(s): de78176

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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."""