pratikshahp commited on
Commit
c1e2ddb
·
verified ·
1 Parent(s): 7e92b1a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -15,7 +15,7 @@ llm = HuggingFaceEndpoint(
15
  repo_id="mistralai/Mistral-7B-Instruct-v0.3",
16
  huggingfacehub_api_token=HF_TOKEN.strip(),
17
  temperature=0.7,
18
- max_new_tokens=300,
19
  )
20
 
21
  # Define state
@@ -49,7 +49,7 @@ def generate_response_node(state: State):
49
  Severity: {state['severity']}
50
  Resources: {resources}
51
 
52
- You are an emergency response assistant. Provide a detailed response plan for the given situation.
53
  """
54
  response = llm(prompt)
55
  return {"recommendation": response}
 
15
  repo_id="mistralai/Mistral-7B-Instruct-v0.3",
16
  huggingfacehub_api_token=HF_TOKEN.strip(),
17
  temperature=0.7,
18
+ max_new_tokens=150,
19
  )
20
 
21
  # Define state
 
49
  Severity: {state['severity']}
50
  Resources: {resources}
51
 
52
+ You are an emergency response assistant. Provide a detailed response plan for the given situation in 100 words.
53
  """
54
  response = llm(prompt)
55
  return {"recommendation": response}