vpcom commited on
Commit
58d18f7
·
1 Parent(s): c4aff14

fix: use the right class for the generated text

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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
- print(stream)
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,