Update main.py

main.py

@@ -41,11 +41,12 @@ def generate(item: Item):
     )
 
     formatted_prompt = format_prompt(f"{item.system_prompt}, {item.prompt}", item.history)
+    print(formatted_prompt)
     stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
     output = ""
 
     for response in stream:
-        print(response.token.text)
+        #print(response.token.text)
         output += response.token.text
     return output