Update app.py
Browse files
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 |
-
|
18 |
-
|
19 |
-
|
20 |
-
|
21 |
-
|
|
|
|
|
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
|