updated streaming code

app.py

@@ -20,7 +20,7 @@ marketing_email_prompt = """Below is a product and description, please write a m
 {}"""
 
 def greet(product,description):
-    output = llm.create_chat_completion(
+    response = llm.create_chat_completion(
     messages=[
         {
             "role": "system",
@@ -39,12 +39,10 @@ def greet(product,description):
     temperature=0.7,
     stream=True
     )
-    for chunk in output:
-        delta = chunk['choices'][0]['delta']
-        if 'role' in delta:
-            yield delta['role']
-        elif 'content' in delta:
-            yield delta['content']
-
+    partial_message = ""
+    for chunk in response:
+        if chunk.choices[0].delta.content is not None:
+            partial_message = partial_message + chunk.choices[0].delta.content
+            yield partial_message
 demo = gr.Interface(fn=greet, inputs=["text","text"], outputs="text")
 demo.launch()