danghungithp commited on
Commit
afaab7f
·
verified ·
1 Parent(s): 25ddfe2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ logging.basicConfig(level=logging.INFO)
11
  logger = logging.getLogger(__name__)
12
 
13
  app = FastAPI()
14
- MODEL_NAME = 'tinyllama'
15
 
16
  def get_llm():
17
  callback_manager = CallbackManager([StreamingStdOutCallbackHandler()])
 
11
  logger = logging.getLogger(__name__)
12
 
13
  app = FastAPI()
14
+ MODEL_NAME = 'deepseek-r1:1.5b'
15
 
16
  def get_llm():
17
  callback_manager = CallbackManager([StreamingStdOutCallbackHandler()])