JustML57 commited on
Commit
e97fb36
·
verified ·
1 Parent(s): ccc1996

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -41,10 +41,11 @@ model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud'
41
  model_id = "deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B"
42
 
43
  model = HfApiModel(
44
- max_tokens=2096,
45
- temperature=0.5,
46
- model_id=model_id,#'Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
47
- custom_role_conversions=None,
 
48
  )
49
 
50
 
 
41
  model_id = "deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B"
42
 
43
  model = HfApiModel(
44
+ max_tokens=2096,
45
+ temperature=0.5,
46
+ model_id=model_id,#'Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
47
+ custom_role_conversions=None,
48
+ token=os.getenv('hf_token')
49
  )
50
 
51