salomonsky commited on
Commit
f27b29a
·
verified ·
1 Parent(s): 770c41b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -53,6 +53,11 @@ def generate(
53
 
54
  formatted_prompt = format_prompt(user_input, history)
55
  stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=True)
 
 
 
 
 
56
  response = ' '.join(response.split()).replace('</s>', '')
57
 
58
  # Mostrar respuesta en un área de texto
 
53
 
54
  formatted_prompt = format_prompt(user_input, history)
55
  stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=True)
56
+
57
+ response = ""
58
+ for response_token in stream:
59
+ response += response_token.token.text
60
+
61
  response = ' '.join(response.split()).replace('</s>', '')
62
 
63
  # Mostrar respuesta en un área de texto