Update app.py
Browse files
app.py
CHANGED
@@ -112,7 +112,7 @@ def generate_text(prompt, system_prompt):
|
|
112 |
{"role": "user", "content": prompt}
|
113 |
]
|
114 |
try:
|
115 |
-
response = client.workers.ai.
|
116 |
account_id=ACCOUNT_ID,
|
117 |
model=CHAT_MODEL,
|
118 |
messages=messages,
|
|
|
112 |
{"role": "user", "content": prompt}
|
113 |
]
|
114 |
try:
|
115 |
+
response = client.workers.ai.with_streaming_response.run(
|
116 |
account_id=ACCOUNT_ID,
|
117 |
model=CHAT_MODEL,
|
118 |
messages=messages,
|