Illia56 commited on
Commit
5018467
1 Parent(s): 9bf9831

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -97,7 +97,7 @@ def predict(
97
  Predict a response using a client.
98
  """
99
  client = Client(model_url)
100
- response = client.predict(message, system_prompt, 0.7, 4096, 0.5, 1.2, api_name="/chat")
101
  return response
102
 
103
  PATH = os.path.join(os.path.expanduser("~"), "Data")
 
97
  Predict a response using a client.
98
  """
99
  client = Client(model_url)
100
+ response = client.predict(message, system_prompt, 0.7, 4096, 0.5, 1.2, api_name=("/chat_1" if model_url== models["Llama2-70b"]["chat_link"] else '/chat'))
101
  return response
102
 
103
  PATH = os.path.join(os.path.expanduser("~"), "Data")