amir22010 commited on
Commit
4571273
·
1 Parent(s): 082d04d

updated streaming code

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -41,8 +41,9 @@ def greet(product,description):
41
  )
42
  partial_message = ""
43
  for chunk in response:
44
- if chunk.choices[0].delta.content is not None:
45
- partial_message = partial_message + chunk.choices[0].delta.content
46
  yield partial_message
 
47
  demo = gr.Interface(fn=greet, inputs=["text","text"], outputs="text")
48
  demo.launch()
 
41
  )
42
  partial_message = ""
43
  for chunk in response:
44
+ if chunk['choices'][0]['delta']['content'] is not None:
45
+ partial_message = partial_message + chunk['choices'][0]['delta'].get('content', '')
46
  yield partial_message
47
+
48
  demo = gr.Interface(fn=greet, inputs=["text","text"], outputs="text")
49
  demo.launch()