Update app.py

app.py

@@ -55,8 +55,15 @@ def generate(prompt, history, system_prompt, temperature=0.9, max_new_tokens=256
 
     stream1 = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
     stream2 = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=False, return_full_text=False)
-    print(f">> STREAM1 - '{stream1}'")
-    print(f">> STREAM2 - '{stream2}'")
+    #print(f">> STREAM1 - '{stream1}'")
+    #print(f">> STREAM2 - '{stream2}'")
+    a, b = ">> STREAM1:", ">> STREAM2:"
+    
+    for i in stream1: a += f"\n   - '{str(i)}'"
+    for i in stream2: b += f"\n   - '{str(i)}'"
+    print(a + "\n")
+    print(b + "\n")
+    
     # Generate text from the HF inference
     stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
     output = ""