Update app.py
Browse files
app.py
CHANGED
|
@@ -214,6 +214,7 @@ def get_window_url_params():
|
|
| 214 |
# app.py
|
| 215 |
@spaces.GPU
|
| 216 |
def generate(
|
|
|
|
| 217 |
message: str,
|
| 218 |
chat_history: list[tuple[str, str]],
|
| 219 |
# system_prompt: str,
|
|
@@ -223,7 +224,9 @@ def generate(
|
|
| 223 |
top_k: int = 50,
|
| 224 |
repetition_penalty: float = 1.2,
|
| 225 |
) -> Iterator[str]: # Change return type hint to Iterator[str]
|
| 226 |
-
|
|
|
|
|
|
|
| 227 |
|
| 228 |
# Construct the input prompt using the functions from the system_prompt_config module
|
| 229 |
input_prompt = construct_input_prompt(chat_history, message)
|
|
|
|
| 214 |
# app.py
|
| 215 |
@spaces.GPU
|
| 216 |
def generate(
|
| 217 |
+
request: gr.Request
|
| 218 |
message: str,
|
| 219 |
chat_history: list[tuple[str, str]],
|
| 220 |
# system_prompt: str,
|
|
|
|
| 224 |
top_k: int = 50,
|
| 225 |
repetition_penalty: float = 1.2,
|
| 226 |
) -> Iterator[str]: # Change return type hint to Iterator[str]
|
| 227 |
+
|
| 228 |
+
params = request.query_params
|
| 229 |
+
print(params)
|
| 230 |
|
| 231 |
# Construct the input prompt using the functions from the system_prompt_config module
|
| 232 |
input_prompt = construct_input_prompt(chat_history, message)
|