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,
|