muryshev commited on
Commit
185e96f
·
verified ·
1 Parent(s): fae4fdc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -33,6 +33,7 @@ def generate(
33
  #stream = client.text_generation(prompt, **generate_kwargs, stream=True, details=False, return_full_text=False)
34
  response = client.text_generation(prompt, **generate_kwargs, stream=False, details=False, return_full_text=False)
35
  print(response)
 
36
  output = ""
37
 
38
  for response in stream:
 
33
  #stream = client.text_generation(prompt, **generate_kwargs, stream=True, details=False, return_full_text=False)
34
  response = client.text_generation(prompt, **generate_kwargs, stream=False, details=False, return_full_text=False)
35
  print(response)
36
+ return response
37
  output = ""
38
 
39
  for response in stream: