Illia56 commited on
Commit
d4138a4
1 Parent(s): df5bc90

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -8,6 +8,8 @@ This Space demonstrates model [Llama-2-70b-chat-hf](https://huggingface.co/meta-
8
  | Model | Llama2 | Llama2-hf | Llama2-chat | Llama2-chat-hf |
9
  |---|---|---|---|---|
10
  | 70B | [Link](https://huggingface.co/meta-llama/Llama-2-70b) | [Link](https://huggingface.co/meta-llama/Llama-2-70b-hf) | [Link](https://huggingface.co/meta-llama/Llama-2-70b-chat) | [Link](https://huggingface.co/meta-llama/Llama-2-70b-chat-hf) |
 
 
11
  """
12
 
13
  # Initialize client
@@ -36,7 +38,7 @@ def predict(message, system_prompt='', temperature=0.7, max_new_tokens=4096,Topp
36
  max_new_tokens, # int | float (numeric value between 0 and 4096)
37
  Topp, # int | float (numeric value between 0.0 and 1)
38
  Repetitionpenalty, # int | float (numeric value between 1.0 and 2.0)
39
- api_name="/chat"
40
  )
41
  st.write("Done")
42
  return response
 
8
  | Model | Llama2 | Llama2-hf | Llama2-chat | Llama2-chat-hf |
9
  |---|---|---|---|---|
10
  | 70B | [Link](https://huggingface.co/meta-llama/Llama-2-70b) | [Link](https://huggingface.co/meta-llama/Llama-2-70b-hf) | [Link](https://huggingface.co/meta-llama/Llama-2-70b-chat) | [Link](https://huggingface.co/meta-llama/Llama-2-70b-chat-hf) |
11
+
12
+ ---
13
  """
14
 
15
  # Initialize client
 
38
  max_new_tokens, # int | float (numeric value between 0 and 4096)
39
  Topp, # int | float (numeric value between 0.0 and 1)
40
  Repetitionpenalty, # int | float (numeric value between 1.0 and 2.0)
41
+ api_name="/chat_1"
42
  )
43
  st.write("Done")
44
  return response