Update app.py
Browse files
app.py
CHANGED
@@ -56,11 +56,12 @@ async def echo(message, history, state, persona):
|
|
56 |
state['messages'] = chat_history
|
57 |
|
58 |
response_chunks = ""
|
|
|
59 |
model_content_stream = await client.models.generate_content_stream(
|
60 |
-
|
61 |
-
|
62 |
-
|
63 |
-
)
|
64 |
async for chunk in model_content_stream:
|
65 |
response_chunks += chunk.text
|
66 |
# when model generates too fast, Gradio does not respond that in real-time.
|
|
|
56 |
state['messages'] = chat_history
|
57 |
|
58 |
response_chunks = ""
|
59 |
+
print(state['messages'])
|
60 |
model_content_stream = await client.models.generate_content_stream(
|
61 |
+
model=args.model,
|
62 |
+
contents=state['messages'],
|
63 |
+
config=types.GenerateContentConfig(seed=args.seed),
|
64 |
+
)
|
65 |
async for chunk in model_content_stream:
|
66 |
response_chunks += chunk.text
|
67 |
# when model generates too fast, Gradio does not respond that in real-time.
|