Spaces:
Runtime error
Runtime error
fix: use the right class for the generated text
Browse files
app.py
CHANGED
|
@@ -137,10 +137,12 @@ def generate(
|
|
| 137 |
|
| 138 |
if len(ENDPOINT)>0:
|
| 139 |
stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
|
|
|
|
|
|
|
| 140 |
else:
|
| 141 |
stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=False, details=True, return_full_text=False)
|
| 142 |
|
| 143 |
-
|
| 144 |
|
| 145 |
output = ""
|
| 146 |
for response in stream:
|
|
@@ -206,7 +208,6 @@ chatbot = gr.Chatbot(label="PersianGPT",
|
|
| 206 |
show_share_button=True,
|
| 207 |
show_copy_button=True,
|
| 208 |
bubble_full_width = False)
|
| 209 |
-
|
| 210 |
textbox = gr.Textbox(
|
| 211 |
label="textbox",
|
| 212 |
container=False,
|
|
|
|
| 137 |
|
| 138 |
if len(ENDPOINT)>0:
|
| 139 |
stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
|
| 140 |
+
print(f"STREAM: {stream}\n")
|
| 141 |
+
stream = stream.generated_text
|
| 142 |
else:
|
| 143 |
stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=False, details=True, return_full_text=False)
|
| 144 |
|
| 145 |
+
|
| 146 |
|
| 147 |
output = ""
|
| 148 |
for response in stream:
|
|
|
|
| 208 |
show_share_button=True,
|
| 209 |
show_copy_button=True,
|
| 210 |
bubble_full_width = False)
|
|
|
|
| 211 |
textbox = gr.Textbox(
|
| 212 |
label="textbox",
|
| 213 |
container=False,
|