Update app.py
Browse files
app.py
CHANGED
|
@@ -200,6 +200,7 @@ def construct_input_prompt(chat_history, message, personalized_data):
|
|
| 200 |
# app.py
|
| 201 |
@spaces.GPU
|
| 202 |
def generate(
|
|
|
|
| 203 |
message: str,
|
| 204 |
chat_history: list[tuple[str, str]],
|
| 205 |
# input_prompt: str,
|
|
@@ -208,7 +209,6 @@ def generate(
|
|
| 208 |
top_p: float = 0.9,
|
| 209 |
top_k: int = 50,
|
| 210 |
repetition_penalty: float = 1.2,
|
| 211 |
-
request: gr.Request, # To fetch query params
|
| 212 |
) -> Iterator[str]: # Change return type hint to Iterator[str]
|
| 213 |
|
| 214 |
conversation = []
|
|
|
|
| 200 |
# app.py
|
| 201 |
@spaces.GPU
|
| 202 |
def generate(
|
| 203 |
+
request: gr.Request, # To fetch query params
|
| 204 |
message: str,
|
| 205 |
chat_history: list[tuple[str, str]],
|
| 206 |
# input_prompt: str,
|
|
|
|
| 209 |
top_p: float = 0.9,
|
| 210 |
top_k: int = 50,
|
| 211 |
repetition_penalty: float = 1.2,
|
|
|
|
| 212 |
) -> Iterator[str]: # Change return type hint to Iterator[str]
|
| 213 |
|
| 214 |
conversation = []
|