Illia56 commited on
Commit
174e2c2
1 Parent(s): 3a0dfb2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -5
app.py CHANGED
@@ -14,11 +14,13 @@ client = Client("https://ysharma-explore-llamav2-with-tgi.hf.space/")
14
  # Prediction function
15
  def predict(message, system_prompt="", temperature=0.9, max_new_tokens=4096):
16
  return client.predict(
17
- message=message,
18
- system_prompt=system_prompt,
19
- temperature=temperature,
20
- max_new_tokens=max_new_tokens,
21
- api_name="/chat"
 
 
22
  )
23
 
24
  # Streamlit UI
 
14
  # Prediction function
15
  def predict(message, system_prompt="", temperature=0.9, max_new_tokens=4096):
16
  return client.predict(
17
+ message, # str in 'Message' Textbox component
18
+ system_prompt, # str in 'Optional system prompt' Textbox component
19
+ temperature, # int | float (numeric value between 0.0 and 1.0)
20
+ max_new_tokens, # int | float (numeric value between 0 and 4096)
21
+ 0.3, # int | float (numeric value between 0.0 and 1)
22
+ 1, # int | float (numeric value between 1.0 and 2.0)
23
+ api_name="/chat"
24
  )
25
 
26
  # Streamlit UI